【蓝桥杯】2015初赛 星系炸弹

题目描述

在X星系的广袤空间中漂浮着许多X星人造“炸弹”，用来作为宇宙中的路标。
每个炸弹都可以设定多少天之后爆炸。
比如：阿尔法炸弹2015年1月1日放置，定时为15天，则它在2015年1月16日爆炸。
有一个贝塔炸弹，a年b月c日放置，定时为n天，请你计算它爆炸的准确日期。

输入

输入存在多组数据，每组数据输入一行，每一行输入四个正整数a,b,c,n
输入保证日期在1000-01-01到2020-01-01之间，且日期合法。
n不超过1000

输出

请填写该日期，格式为 yyyy-mm-dd 即4位年份2位月份2位日期。比如：2015-02-19
请严格按照格式书写。不能出现其它文字或符号。

样例输入

2015 1 1 15
2014 11 9 1000

样例输出

2015-01-16
2017-08-05

n比较小直接枚举复杂度O(n)可过

y1 = [1,3,5,7,8,10,12]
y2 = [4,6,9,11]
def rn(n):
    if (n%4==0 and n%100!=0) or n%400==0:
        return True
    else:
        return False
while True:

    n,y,r,t = map(int,input().split())
    for i in range(t):
        r+=1
        if (y in y1)and r>31:
            r=1
            y+=1
        elif (y in y2)and r>30:
            r=1
            y+=1
        elif y==2 and rn(n) and r>29:
            r=1
            y+=1
        elif y==2  and not  rn(n) and r>28:
            r=1
            y+=1
        if y==13:
            y=1
            n+=1
    print('%04d-%02d-%02d' % (n , y , r))