finds next date for a given date

Author: NirmalSilwal

Udacity Data Structure and Algorithms Nanodegree/Introduction/FindNextDay.py

@@ -0,0 +1,49 @@
+###
+### Define a simple nextDay procedure, that assumes
+### every month has 30 days.
+###
+### For example:
+###    nextDay(1999, 12, 30) => (2000, 1, 1)
+###    nextDay(2013, 1, 30) => (2013, 2, 1)
+###    nextDay(2012, 12, 30) => (2013, 1, 1)  (even though December really has 31 days)
+###
+
+
+def next_day2(year, month, day):
+    """Simple version: assume every month has 30 days"""
+    if day < 30:
+        return year, month, day + 1
+    else:
+        if month == 12:
+            return year + 1, 1, 1
+        else:
+            return year, month + 1, 1
+
+
+def next_day(year, month, day):
+    """
+    Returns the year, month, day of the next day.
+    Simple version: assume every month has 30 days.
+    """
+    # YOUR CODE HERE
+    if month == 12:
+        if day == 30:
+            day = 1
+            year += 1
+            month = 1
+        else:
+            day += 1
+
+    else:
+        if day == 30:
+            month += 1
+            day = 1
+
+        else:
+            day += 1
+
+    return year, month, day
+
+
+y, m, d = next_day(2012, 12, 1)
+print(y, m, d)