题目描述
国王将金币作为工资,发放给忠诚的骑士。第一天,骑士收到一枚金币;之后两天(第二天和第三天),每天收到两枚金币;之后三天(第四、五、六天),每天收到三枚金币;之后四天(第七、八、九、十天),每天收到四枚金币……;这种工资发放模式会一直这样延续下去:当连续 n n n 天每天收到 n n n 枚金币后,骑士会在之后的连续 n + 1 n+1 n+1 天里,每天收到 n + 1 n+1 n+1 枚金币。
请计算在前 k k k 天里,骑士一共获得了多少金币。
输入格式
一个正整数 k k k,表示发放金币的天数。
输出格式
一个正整数,即骑士收到的金币数。
输入输出样例
输入
6
输出
14
说明/提示
【样例 1 说明】
骑士第一天收到一枚金币;第二天和第三天,每天收到两枚金币;第四、五、六天,每天收到三枚金币。因此一共收到 1 + 2 + 2 + 3 + 3 + 3 = 14 1+2+2+3+3+3=14 1+2+2+3+3+3=14 枚金币。
对于 100 % 100\% 100% 的数据, 1 ≤ k ≤ 1 0 4 1\le k\le 10^4 1≤k≤104。
方式
代码
class Solution:@staticmethoddef oi_input():"""从标准输入读取数据"""day = int(input())return day@staticmethoddef oi_test():"""提供测试数据"""return 6@staticmethoddef solution(day):sum, feature = 0, 0while day > feature * (feature + 1) // 2:feature += 1full_days = feature - 1remain_day = day - full_days * (full_days + 1) // 2for i in range(1, full_days + 1):sum += i * iprint(sum + remain_day * feature)oi_input = Solution.oi_input
oi_test = Solution.oi_test
solution = Solution.solutionif __name__ == '__main__':day = oi_test()# day = oi_input()solution(day)