時空限制    1000ms/128MB

題目描述

國王將金幣作為工資，發放給忠誠的騎士。第一天，騎士收到一枚金幣；之後兩天（第二天和第三天），每天收到兩枚金幣；之後三天（第四、五、六天），每天收到三枚金幣；之後四天（第七、八、九、十天），每天收到四枚金幣……；這種工資發放模式會一直這樣延續下去：當連續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;
}