洛谷题单3-P2669 [NOIP 2015 普及组] 金币-python-流程图重构

题目描述

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

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

输入格式

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

输出格式

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

输入输出样例

输入

6

输出

14

说明/提示

【样例 1 说明】

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

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

方式

代码

class Solution:
    @staticmethod
    def oi_input():
        """从标准输入读取数据"""
        day = int(input())
        return day

    @staticmethod
    def oi_test():
        """提供测试数据"""
        return 6

    @staticmethod
    def solution(day):
        sum, feature = 0, 0

        while day > feature * (feature + 1) // 2:
            feature += 1

        full_days = feature - 1
        remain_day = day - full_days * (full_days + 1) // 2

        for i in range(1, full_days + 1):
            sum += i * i
        print(sum + remain_day * feature)


oi_input = Solution.oi_input
oi_test = Solution.oi_test
solution = Solution.solution

if __name__ == '__main__':
    day = oi_test()
    # day = oi_input()
    solution(day)

流程图