题目描述
国王将金币作为工资,发放给忠诚的骑士。第一天,骑士收到一枚金币;之后两天(第二天和第三天),每天收到两枚金币;之后三天(第四、五、六天),每天收到三枚金币;之后四天(第七、八、九、十天),每天收到四枚金币......;这种工资发放模式会一直这样延续下去:当连续N天每天收到N枚金币后,骑士会在之后的连续N+1天里,每天收到N+1枚金币。
请计算在前K天里,骑士一共获得了多少金币。
输入描述
只有1行,包含一个正整数K,表示发放金币的天数。
输出描述
只有1行,包含一个正整数,即骑士收到的金币数。
样例输入 1
6
样例输出1
14
样例输入2
1000
样例输出2
29820
思路解析:
转化成数学知识就很好计算了,具体为:
1、k算作和,就是数列1,2,3,4...的和,当这个数列的前n项和正好大于等于k,记下这个n(代码中为index)以及这个和sum;
2、然后计算金币数,即是前n-1项1^2,2^2,3^3,4^4......(n-1)^2的和,记下为gold;
3、用gold + (k - (sum - index) ) * index,这才是最终的gold
代码如下:
#C0224 [2015普及组-A]金币.py
k = eval(input())
Sum = 0
index = 0
gold = 0
for i in range(1,k+1): #计算和,记下Sum>=k的i值,一定从1开始加到k(k也得能加进去)
Sum += i
if Sum >= k:
index = i
break
for j in range(1,index): #计算金币数
gold += j * j
gold += index * (k - (Sum - index))#关键读懂
print(gold)