This problem is in Contest(s) below:
Contest 1063:2023-WE 百题大战03：分支结构
Contest 1038:Python-第6章-函数