ybt1969 金币（NOIP2015普及组第1题）

ybt1969  金币（NOIP2015普及组第1题）

时空限制    1000ms/64MB

题目描述

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

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

输入

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

输出

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

输入样例1

6

输出样例1

14

输入样例2

1000

输出样例2

29820

说明

        【输入输出样例 1 说明】骑士第一天收到一枚金币；第二天和第三天，每天收到两枚金币；第四、五、六天，每天收到三枚金币。因此一共收到 1+2+2+3+3+3=14 枚金币。

        对于 100%的数据， 1 ≤ K ≤ 10,000。

代码

#include<iostream>
using namespace std;

int main(){
    int k,ans=0;
    cin>>k;
    for (int i=1; k>0; ++i)
        if (k>=i) ans+=i*i,k-=i;
        else ans+=i*k,k=0;
    cout<<ans<<endl;
    return 0;
}