【15NOIP普及组】金币

http://ybt.ssoier.cn:8088/problem_show.php?pid=1969
【题目描述】
国王将金币作为工资，发放给忠诚的骑士。第一天，骑士收到一枚金币；之后两天（第二天和第三天），每天收到两枚金币；之后三天（第四、五、六天），每天收到三枚金币；之后四天（第七、八、九、十天），每天收到四枚金币……；这种工资发放模式会一直这样延续下去：当连续N天每天收到N枚金币后，骑士会在之后的连续N+1天里，每天收到N+1枚金币。

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

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

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

【输入样例】
6
【输出样例】
14
【提示】
样例测试点#2

输入：

1000

输出：

29820

本题就是可以理解为i个数的平方的和，但是最后是否还是i的平方，就取决于n-i是否刚好是0？下面直接上代码！

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