第五题:金币
题目描述
国王将金币作为工资,发放给忠诚的骑士。第一天,骑士收到一枚金币;之后两天(第二天和第三天),每天收到两枚金币;之后三天(第四、五、六天),每天收到三枚金币;之后四天(第七、八、九、十天),每天收到四枚金币......;这种工资发放模式会一直这样延续下去:当连续 N 天每天收到 N 枚金币后,骑士会在之后的连续N+1 天里,每天收到 N+1 枚金币。
请计算在前 K 天里,骑士一共获得了多少金币。
输入描述
输入只有 1 行,包含一个正整数K (1≤K≤104),表示发放金币的天数。
输出描述
输出只有 1 行,包含一个正整数,即骑士收到的金币数。
输入输出样例
示例 1
输入
6
输出
14
示例 2
输入
1000
输出
29820
运行限制
- 最大运行时间:1s
- 最大运行内存: 256M
代码如下:
k=int(input())
i=1
N=0
while k>=i:
k-=i
N+=i*i
i+=1
if k!=0:
N+=k*i
print(N)