2991: [NOIP2015]金币 T1

Question

题目描述

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

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

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

输出只有1行，包含一个正整数，即骑士收到的金币数。
样例输入
6
样例输出
14
提示

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

【数据说明】
对于100%的数据，1 ≤ K ≤ 10,000。

来源

NOIP2015普及组
有代码，答案错误，80%，望回答

#include<bits/stdc++.h>
using namespace std;
int a[3][2];
int main() {
    int n,sum=0,k,k_2;
    cin >> n;
    for(int i=1;;i++){
        int sum_2=sum;
        sum+=i*i;
        if(sum>n || sum==n)
        {
            k=i;
            k_2=sum_2;
            break;
        }
    }
    //cout<<sum<<endl<<k<<endl<<k_2;
    cout<<sum+(n-k_2-1)*(k+1);
    return 0;
    }
 

Answer 1

该回答引用GPTᴼᴾᴱᴺᴬᴵ
该代码思路是按照题目的金币发放规律计算，但是有一些问题。

首先，在循环计算i天每天获得的金币数时，代码中使用的是i * i，这是错误的，应该是按照题目规律来计算，即当连续N天每天收到N枚金币后，骑士会在之后的连续N+1天里，每天收到N+1枚金币。所以应该使用i <= k ? i : k + 1来计算。

其次，代码中的变量a并没有用到，可以删除。

最后，对于边界条件的判断，可以使用sum <= n来判断是否超过前K天发放的金币数。

下面是修改后的代码，可以AC该题。

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int k = scanner.nextInt();
        int coins = 0;
        int days = 0;
        int n = 1;
        while (days < k) {
            for (int i = 0; i < n && days < k; i++) {
                coins += n;
                days++;
            }
            n++;
        }
        System.out.println(coins);
    }

}