NOJ南邮上机 忠诚的骑士 PROB1011 Python

PROB1011   忠诚的骑士

描述：

一位忠诚的骑士在战场上立下汗马功劳。

回到王都后，经济拮据的国王便许诺，将金币作为工资，发放给这位骑士，而骑士每天获得的金币会随着时间而增加。

第一天骑士收到一枚金币；之后两天（第二天和第三天），每天收到两枚金币；之后三天（第四、五、六天），每天收到三枚金币；之后四天，每天收到四枚金币，以此类推；这种工资发放模式会一直延续下去，当连续N天收到N枚金币后，骑士会在之后的N+1天，每天收到N+1枚金币。

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

输入：

输入仅包含一个正整数K，表示发放金币的天数，满足1≤K≤10000。

输出：

输出仅包含一个正整数，即骑士收到的金币数。

样例输入：

6

样例输出： 

14

 基于python实现的代码：

def calculate_gold(K):
    total_gold = 0
    current_day = 1
    cycle_length = 1
    while K > 0:
        if K >= cycle_length:
            total_gold += cycle_length * current_day
            K -= cycle_length
        else:
            total_gold += current_day * K
            break
        cycle_length += 1
        current_day += 1
    return total_gold
K =  int(input().strip())
print(calculate_gold(K))