金币 NOIP2015普及组

【题目描述】

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

请计算在前K天里，骑士一共获得了多少金币。

【输入】

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

【输出】

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

【输入样例】

6

【输出样例】

14

【提示】

样例测试点#2

输入：

1000

输出：

29820

用n模拟发过几轮金币，计算出共发几轮和还剩几天

代码如下

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int k,n=1,ans=0;
	cin>>k;
	while(k>=n)
	{
		k-=n;
		n++;
	}
	n--;
	for(int i=1;i<=n;i++)
		ans+=i*i;
	ans+=(n+1)*k;
	cout<<ans;
	return 0;
}