题目描述
国王将金币作为工资，发放给忠诚的骑士。第一天，骑士收到一枚金币；之后两天（第二天和第三天），每天收到两枚金币；之后三天（第四、五、六天），每天收到三枚金币；之后四天（第七、八、九、十天），每天收到四枚金币……；这种工资发放模式会一直这样延续下去：当连续N天每天收到N枚金币后，骑士会在之后的连续N+1天里，每天收到N+1枚金币。
请计算在前K天里，骑士一共获得了多少金币。
输入格式
一个正整数K，表示发放金币的天数。
输出格式
一个正整数，即骑士收到的金币数。
输入输出样例
输入 #1
6
输出 #1
14
输入 #2
1000
输出 #2
29820
说明/提示
【输入输出样例 1 说明】
骑士第一天收到一枚金币；第二天和第三天，每天收到两枚金币；第四、五、六天，每天收到三枚金币。因此一共收到 1+2+2+3+3+3=14 枚金币。
对于 100%的数据， 1≤K≤10,000。
代码：
#include<bits/stdc++.h>
using namespace std;
int main()
{
int sum = 0;
double n;
cin >> n;
if(sqrt(1 + 8 * n) == 1.0 * int(sqrt(1 + 8 * n)))
{
int k = (sqrt(1 + 8 * n) - 1) / 2;
cout << (k * (k + 1) * (2 * k + 1)) / 6;
}
if(sqrt(1 + 8 * n) != 1.0 * int(sqrt(1 + 8 * n)))
{
int k = (sqrt(1 + 8 * n) - 1) / 2;
sum += (k * (k + 1) * (2 * k + 1)) / 6;
sum += (n - (k *(k + 1) / 2)) * (k + 1);
cout << sum;
}
return 0;
}