Python实现欧拉计划19题

You are given the following information, but you may prefer to do some research for yourself.

• 1 Jan 1900 was a Monday.
• Thirty days has September,
  April, June and November.
  All the rest have thirty-one,
  Saving February alone,
  Which has twenty-eight, rain or shine.
  And on leap years, twenty-nine.
• A leap year occurs on any year evenly divisible by 4, but not on a century unless it is divisible by 400.

How many Sundays fell on the first of the month during the twentieth century (1 Jan 1901 to 31 Dec 2000)?

--------------------------------------------------------------------------------------------------------------------------------------------

def leap(x):
    if x%100==0 and x%400==0:
        return 1
    elif x%100!=0 and x%4==0:
        return 1
    else:
        return 0


i=0
sum=0
month=[[31,28,31,30,31,30,31,31,30,31,30,31],[31,29,31,30,31,30,31,31,30,31,30,31]]
if leap(1900):
    sum=sum+366
else:
    sum=sum+365
for n in range(1901,2001):
    if leap(n):
        for mon in range(0,12):
            sum=sum+month[1][mon]
            if(sum%7==6):
                i=i+1
    else:
        for mon in range(0,12):
            sum=sum+month[0][mon]
            if(sum%7==6):
                i=i+1
if sum%7==6:
    i=i-1
print i