南邮NOJ上机系统#PROB1011忠诚的骑士

忠诚的骑士

描述：
一位忠诚的骑士在战场上立下汗马功劳。

回到王都后，经济拮据的国王便许诺，将金币作为工资，发放给这位骑士，而骑士每天获得的金币会随着时间而增加。

第一天骑士收到一枚金币；之后两天（第二天和第三天），每天收到两枚金币；之后三天（第四、五、六天），每天收到三枚金币；之后四天，每天收到四枚金币，以此类推；这种工资发放模式会一直延续下去，当连续N天收到N枚金币后，骑士会在之后的N+1天，每天收到N+1枚金币。

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

输入：
输入仅包含一个正整数K，表示发放金币的天数，满足1≤K≤10000。

输出：
输出仅包含一个正整数，即骑士收到的金币数。

样例输入：
6
样例输出：
14
样例输入：
1000
样例输出：
29820
注释：
对于第一个样例，骑士一共收到1+2+2+3+3+3=14枚金币。

C++代码

#include<iostream>

int main(){
    int k,n,count,more;
    std::cin>>k;
    n=1;
    count=0;
    while(count<k){    //k天看作1+2+…+n+(n+1-more)天
        count+=n;
        n++;
    }
    more=count-k;    //多算了more天
    count=0;
    for(int i=1;i<n;i++){
        count+=i*i;
    }
    count-=more*(n-1);    //减去多算的more天的金币
    std::cout<<count;
}