【Python】Comet OJ C0224 [2015普及组-A]金币

题目描述

国王将金币作为工资,发放给忠诚的骑士。第一天,骑士收到一枚金币;之后两天(第二天和第三天),每天收到两枚金币;之后三天(第四、五、六天),每天收到三枚金币;之后四天(第七、八、九、十天),每天收到四枚金币......;这种工资发放模式会一直这样延续下去:当连续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)

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章