【问题描述】
国王为他的忠诚的骑士支付金币。在他服役的第一天,骑士收到一枚金币。在接下来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;
}