Python 工程题库 获得昨天和明天的日期

问题

编写一个函数，接收一个日期输入，并输出该日期前一天和后一天的日期‪‬‪‬‪‬‪‬‪‬‮‬‫‬‪‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‫‬

日期格式为：2018-03-21‪‬‪‬‪‬‪‬‪‬‮‬‫‬‪‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‪‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‭‬‪‬‪‬‪‬‪‬‪‬‮‬‫‬‫‬

讨论

可以使用标准数学运算来操纵 datetime 示例。如果想表示一个时间间隔，可以使用 datetime.timedelta 方法。

输入

2018-01-23
1998-01-01
2018-12-31

输出

前一天: 2018-01-22
后一天: 2018-01-24
前一天: 1997-12-31
后一天: 1998-01-02
前一天: 2018-12-30
后一天: 2019-01-01

AC Code

import datetime
import sys


def next_day(date_str):
    timeArray = datetime.datetime.strptime(date_str, "%Y-%m-%d")
    nd = (timeArray + datetime.timedelta(days=1))
    return nd.strftime("%Y-%m-%d")


def prev_day(date_str):
    timeArray = datetime.datetime.strptime(date_str, "%Y-%m-%d")
    pd = (timeArray - datetime.timedelta(days=1))
    return pd.strftime("%Y-%m-%d")


while True:
    line = sys.stdin.readline()
    line = line.strip()
    if line == '':
        break
    print('前一天:', prev_day(line))
    print('后一天:', next_day(line))