【算法题】金币问题

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

解题思路：

  若每次金币都能持续完，则获得金币为n*n，否则是k*n（k为剩余天数）

#include
using namespace std;
int main(){
	int n;  //第n天 
	int coin = 1;  //第1天给的金币
	int keep = 1;  //持续天数 
	int sum = 0;
	cin>>n;
	n -= 1;
	while(n>=0){  //如果第n天持续完 
		sum += coin*coin;
		keep++;
		coin++;  //金币增加 
		n -= keep;
	}
	sum += coin*(n+keep);  //keep无法持续完，加上剩下天数的 
	cout<