NOIP2015 普及组 T1 国王给骑士的金币 Python

[NOIP2015 普及组] 金币

题目背景

NOIP2015 普及组 T1

题目描述

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

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

输入格式

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

输出格式

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

样例 #1

样例输入 #1

6

样例输出 #1

14

样例 #2

样例输入 #2

1000

样例输出 #2

29820

提示

【样例 1 说明】

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

对于 $100\%$ 的数据，$1\le k\le 10^4$。

k = int(input())#获取第几天
count,sum = 0,0
i=0
#最外层循环是控制层数，内层循环是控制每一层的个数
#在设置一个计数的count 记录是否达到指定的天数
while count!=k :
    x = 1
    i += 1
    while x <= i:
        x+=1
        count+=1
        sum += i
        if count == k:
            break
print(sum)