2110. [NOIP2015普及]金币

★   输入文件：2015coin.in   输出文件：2015coin.out   简单对比

时间限制：1 s   内存限制：256 MB

【题目描述】

国王将金币作为工资，发放给忠诚的骑士。第一天骑士收到一枚金币；之后两天（第二天和第三天），每天收到两枚金币；之后三天（第四、五、六天），每天收到三枚金币；之后四天（第七、八、九、十天），每天收到四枚金币……；这种工资发放模式会一直这样延续下去：当连续N天每天收到N枚金币后，骑士会在之的连续N+1天里，每天收到N+1枚金币。请计算在前K天里，骑士一共获得了多少金币。

【输入格式】

输入文件只有 1 行，包含一个正整数 K，表示发放金币的天数。

【输出格式】

输出文件只有 1 行，包含一个正整数，即骑士收到的金币数。

【样例输入】

6

【样例输出】

14

【提示】

【输入出样例1说明】

骑士第一天收到一枚金币；第二和第三天， 每两天收到两枚金币；第四、五、六天，每天收到三枚金币。因此一共收到1+2+2+3+3+3=14枚金币。

【数据说明】

对于 100%的数据，1 ≤ K  ≤ 10,000。

【来源】

NOIP 2015 普及组 第一题

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>

using namespace std;

long long answer;
int tot;

inline int read()
{
	int x=0;int f=1;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
	return x*f;
}

int main()
{
	freopen("2015coin.in","r",stdin);
	freopen("2015coin.out","w",stdout);
	int n=read();
	for(int i=1;;i++)	
	{
		if(tot+i>n)
		{
			answer+=(n-tot)*i;
			printf("%lld",answer);
			return 0;
		}
		answer+=i*i;
		tot+=i;
		if(tot==n)
		{
			printf("%lld",answer);
			return 0;
		}
	}
}