【编程题】【未知出处】输入两个日期计算两个日期相差的天数

【难度：easy】

# 输入为
# y1
# y2
# m1
# m2
# d1
# d2

days1 = [31,29,31,30,31,30,31,31,30,31,30,31]
days2 = [31,28,31,30,31,30,31,31,30,31,30,31]

def countday(year):
	if isday1(year): 
		return sum(days1)
	return sum(days2)

def isday1(y):
	if year % 400 == 0 or (year % 100 != 0 and year % 4 == 0):
		return True
	return False

def backday(y1, m1, d1):
	if isday1(y1):
		return sum(days1[m1+1:]) + days1[m1] - d1
	return sum(days2[m1+1:]) + days2[m1] - d1

def frontday(y1, m1, d1):
	if isday1(y1):
		return sum(days1[:m1+1]) + d1
	return sum(days2[:m1+1]) + d1

def count(y1, m1, d1, y2, m2, d2):
	if isday1(y1): # 是闰年
		return days1[m1+1:m2] + days1[m1] - d1 + d2
	return days2[m1+1:m2] + days2[m1] - d1 + d2

if y2 - y1 == 1:
	return backday(y1, m1, d1) + frontday(y2, m2, d2)
elif y2 - y1 == 0:
	return count(y1, m1, d1, y2, m2, d2)
else:
	sumnum = backday(y1, m1, d1) + frontday(y2, m2, d2)
	for year in range(y1+1, y2):
		sumnum += countday(year)
	return sumnum