달력 수정

src/kr/co/swh/lecture/algorithm/Lv5-4.py

@@ -1,3 +1,35 @@
+def totalDayFromCalendar(year, month, day):
+    dayOfMonth = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30]
+    i=1
+    totaldays = 365 * (int(year) - 1)
+    while i<year:
+        if i % 4 == 0 and i % 100 != 0 or i % 400 == 0:
+            totaldays+=1
+        i += 1
+    premonth = month - 1
+    for b in range(len(dayOfMonth)):
+        if premonth >= (b+1):
+            totaldays += dayOfMonth[b]
+    if month > 2 and year % 4 == 0 and year % 100 != 0 or year % 400 == 0:
+            totaldays +=1
+    totaldays +=1
+    totaldays = totaldays + day
+    return totaldays
+
+def endDayFromTotalDay(year, month):
+    if month == 2:
+        lastday = 28
+        if year % 4 == 0 and year % 100 != 0 or year % 400 == 0:
+            lastday = 29
+        else:
+            lastday = 28
+    else:
+        if month == 4 or month == 6 or month == 9 or month == 11:
+            lastday = 30
+        else:
+            lastday = 31
+    return lastday
+
 def calendar_month(year, month):
     a = totalDayFromCalendar(year, month, 1)
     k = endDayFromTotalDay(year,month)