作业训练二编程题5. 金币

【问题描述】

国王为他的忠诚的骑士支付金币。在他服役的第一天，骑士收到一枚金币。在接下来2天（第二天和第三天的服务），骑士每天收到2金币。在未来三天（第五，第四，和第六天的服务），骑士每天收到三金币。在未来四天（第七，第八，第九，和第十天的服务），骑士每天收到四金币。这一模式的付款方式将继续下去：在接下来的n天骑士每天将收到n枚金币，而在接接下来的n+1天每天将收到n+1枚金币，这里n是正整数。你的程序将确定在任何给定的天数（从第1天开始）支付给骑士的金币总数。

【输入形式】

输入包含至少一行，但不超过21行。输入的每一行包含一个测试案例的数据，即一个整数（1~10000），代表天数。

【输出形式】

每一行输出对应一个测试用例，由天数和支付给骑士的金币总数量组成，中间用空格分隔。

【样例输入】

10
6
10000
1000
21
22

【样例输出】

10 30
6 14
10000 942820
1000 29820
21 91
22 98

【思路分析】

采用数学方法推导，可避免迭代带来的线性时间复杂度。

对于单个数字，提供的代码的时间复杂度为。

#include <iostream>
#include <valarray>

using namespace std;

int main() {
    int n;
    while (cin >> n) {
        int day = sqrt(n * 2 + 0.25) - 0.5, diff = day + 1 - n + day * (day + 1) / 2;
        if (diff == day + 1) cout << n << " " << day * (day + 1) * (2 * day + 1) / 6 << endl;
        else cout << n << " " << (day + 1) * (day + 2) * (2 * day + 3) / 6 - (day + 1) * diff << endl;
    }
    return 0;
}