CSP201509-2 日期计算 (Python)

题目

试题编号：	201509-2
试题名称：	日期计算
时间限制：	1.0s
内存限制：	256.0MB

问题描述
　　给定一个年份y和一个整数d，问这一年的第d天是几月几日？
　　注意闰年的2月有29天。满足下面条件之一的是闰年：
　　1） 年份是4的整数倍，而且不是100的整数倍；
　　2） 年份是400的整数倍。
　　
输入格式
　　输入的第一行包含一个整数y，表示年份，年份在1900到2015之间（包含1900和2015）。
　　输入的第二行包含一个整数d，d在1至365之间。
　　
输出格式
　　输出两行，每行一个整数，分别表示答案的月份和日期。
　　
样例输入
　　2015
　　80
　　
样例输出
　　3
　　21
　　
样例输入
　　2000
　　40
　　
样例输出
　　2
　　9

程序代码

# 输入年份和天数
year = int(input())
n = int(input())

# 闰年标记
flag = False

# 判断是否为闰年
if (year%4 == 0 and year%100 != 0) or year%400 == 0:
    flag = True

# 每月天数列表(除去二月且下标索引从 1 开始)
daysInYear = [0,31,0,31,30,31,30,31,31,30,31,30,31]

# 考虑二月天数加入列表
if flag:
    daysInYear[2] = 29
else:
    daysInYear[2] = 28

# 初始化月份
month = 1

# 循环遍历月份
while True:
    # 当剩余天数大于本月天数(可以到达下月)
    if n - daysInYear[month] > 0:
        n -= daysInYear[month]
        month += 1
    # 当剩余天数就在本月中
    else:
        day = n
        break

# 输出月份和日期
print(month)
print(day)