洛谷题单2-P5716 【深基3.例9】月份天数-python-流程图重构

题目描述

输入年份和月份，输出这一年的这一月有多少天。需要考虑闰年。

输入格式

输入两个正整数，分别表示年份 $y$ 和月数 $m$，以空格隔开。

输出格式

输出一行一个正整数，表示这个月有多少天。

输入输出样例

输入

1926 8

输出

31

说明/提示

数据保证 $1583\le y\le 2020$，$1\le m\le 12$。

方式-先判断是不是闰年

代码

class Solution:
    @staticmethod
    def oi_input():
        """从标准输入读取数据"""
        y, m = map(int, input().split())
        return y, m

    @staticmethod
    def oi_test():
        """提供测试数据"""
        return 1926, 8

    @staticmethod
    def solution(y, m):
        is_leap_year = True if y % 400 == 0 or y % 4 == 0 and y % 100 != 0 else False

        if m == 2:
            if is_leap_year:
                print(29)
            else:
                print(28)
        elif m in [1, 3, 5, 7, 8, 10, 12]:
            print(31)
        else:
            print(30)


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

if __name__ == '__main__':
    y, m = oi_test()
    # y, m = oi_input()
    solution(y, m)

流程图

方式-内置模块的函数

代码

class Solution:
    @staticmethod
    def oi_input():
        """从标准输入读取数据"""
        y, m = map(int, input().split())
        return y, m

    @staticmethod
    def oi_test():
        """提供测试数据"""
        return 1926, 8

    @staticmethod
    def solution(y, m):
        '''calendar.monthrange 函数 返回元组 (该月第一天的星期, 该月天数)'''
        import calendar

        print(calendar.monthrange(y, m)[1])


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

if __name__ == '__main__':
    y, m = oi_test()
    # y, m = oi_input()
    solution(y, m)

流程图