LeetCode 1154 一年中的第几天

class Solution:
    def dayOfYear(self, date: str) -> int:
        months = [0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334]
        time = date.split('-')
        year = int(time[0])
        month = int(time[1])
        day = int(time[2])
        flag = 0
        if year % 400 == 0 and year % 100 != 0 or year % 4 == 0:
            if month>=3:
                flag = 1
        return months[month - 1] + day + flag