L1-024 后天(Python3)

如果今天是星期三，后天就是星期五；如果今天是星期六，后天就是星期一。我们用数字1到7对应星期一到星期日。给定某一天，请你输出那天的“后天”是星期几。

输入格式：

输入第一行给出一个正整数D（1 ≤ D ≤ 7），代表星期里的某一天。

输出格式：

在一行中输出D天的后天是星期几。

输入样例：

3

输出样例：

5

def get_next_day(current_day):
    next_day = (current_day + 2) % 7
    if next_day == 0:
        next_day = 7
    return next_day


D = int(input())
next_day = get_next_day(D)
print(next_day)

 二次优化

def get_next_day(current_day):
    return (current_day + 2) % 7 or 7

在Python中，逻辑运算符`or`的工作方式是，如果第一个表达式的值为真（非零），则返回第一个表达式的值；否则，返回第二个表达式的值。