洛谷P2669 金币（思路+代码详解）Python实现

题目描述

国王将金币作为工资，发放给忠诚的骑士。第一天，骑士收到一枚金币；之后两天（第二天和第三天），每天收到两枚金币；之后三天（第四、五、六天），每天收到三枚金币；之后四天（第七、八、九、十天），每天收到四枚金币……；这种工资发放模式会一直这样延续下去：当连续 n 天每天收到 n 枚金币后，骑士会在之后的连续n+1 天里，每天收到 n+1 枚金币。

请计算在前 k 天里，骑士一共获得了多少金币。

输入格式

一个正整数 k，表示发放金币的天数。

输出格式

一个正整数，即骑士收到的金币数。

输入输出样例

输入 #1复制

6

输出 #1复制

14

输入 #2复制

1000

输出 #2复制

29820

说明/提示

【样例 1 说明】

骑士第一天收到一枚金币；第二天和第三天，每天收到两枚金币；第四、五、六天，每天收到三枚金币。因此一共收到 1+2+2+3+3+3=14 枚金币。

对于 100%100% 的数据，1≤k≤10^4。

 题目分析：主要有以下3点：

1.天数由金币数决定：比如发1枚金币就持续1天发一枚金币，接着金币数递增，发2枚金币就持续2天发两枚金币，故如发n枚金币就持续n天发n枚金币

2.输入发金币的天数符合等差数列之和(但不一定刚好符合)，如：1+2+2+3+3+3=14,输入的6=1+2+3，但如果输入7的话1+2+2+3+3+3+4=18，但发4枚金币就持续了一天，所以要考虑这种情况

3.计算方式：1+2+2+3+3+3=14<=>1^2+2^2+3^2=14

m=int(input())#m作为输入的天数
s=0#s用来记录金币的总数量
n=0#递推每天发金币的数量
while int(n*(1+n)/2)<m:#用int(n*(1+n)/2)来判断n递推到的每天发几个金币
    n+=1#更新n的值
for i in range(1,n+1):#while循环后的int(n*(1+n)/2)是大于等于m并且一定会到发金币到最后的n，但最后的n不一定完全发n天，只有当int(n*(1+n)/2)==m时最后的一个n才发n天
    if i!=n:
        s=s+i**2
    else:当到最后一个i==n时，用i**2-发的几天的金币（int(n*(1+n)/2-m)为多发金币的天数）
        s+=(i**2-int(n*(1+n)/2-m)*i)
print(s)

加油！！!