Python:经典算法题之金币

1、题目描述

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

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

（1）输入描述

输入只有 1 行，包含一个正整数 K （1≤K≤10^4），表示发放金币的天数。

（2）输出描述

输出只有 1 行，包含一个正整数，即骑士收到的金币数。

  (3) 分析

---

 方法一：

import os
import sys
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)

方法二：

n=int(input())
d=0
sum=0
for i in range(1,1000):
   d+=int(i)
   if d<=n:
      sum+=i*i
   else:
      j=d-n
      sum=sum+i*(i-j)
      break
print(sum)