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

题目描述

国王将金币作为工资,发放给忠诚的骑士。第一天,骑士收到一枚金币;之后两天(第二天和第三天),每天收到两枚金币;之后三天(第四、五、六天),每天收到三枚金币;之后四天(第七、八、九、十天),每天收到四枚金币......;这种工资发放模式会一直这样延续下去:当连续 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。

方式

代码

python 复制代码
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)

流程图

是 否 开始 输入天数day 初始化sum=0, feature=0 day > feature*(feature+1)/2 ? feature += 1 计算full_days = feature-1 计算remain_day = day - full_days*(full_days+1)/2 循环i=1到full_days: sum += i² sum += remain_day * feature 输出sum 结束

相关推荐
KangkangLoveNLP6 分钟前
手动实现一个迷你Llama:使用SentencePiece实现自己的tokenizer
人工智能·深度学习·学习·算法·transformer·llama
独好紫罗兰10 分钟前
洛谷题单3-P1420 最长连号-python-流程图重构
开发语言·python·算法
kingmax5421200814 分钟前
深入解析:使用Python爬取Bilibili视频
开发语言·python·音视频
申雪菱25 分钟前
Scheme语言的数据挖掘
开发语言·后端·golang
程序员一诺29 分钟前
【Flask开发】嘿马文学web完整flask项目第1篇:简介【附代码文档】
后端·python·flask·框架
Aerkui34 分钟前
Python列表(List)深度解析
开发语言·python
柯ran34 分钟前
数据结构|排序算法(一)快速排序
数据结构·算法·排序算法
Bruce_Liuxiaowei36 分钟前
基于Flask的MBA考生成绩查询系统设计与实现
后端·python·flask
啊阿狸不会拉杆37 分钟前
第二十章:Python-Matplotlib库实现函数可视化
开发语言·python·matplotlib
pipip.39 分钟前
搜索二维矩阵
数据结构·算法·矩阵