Python洛谷题目P5716合法性判断

Python判断洛谷题目号P5716合法性

题目描述： 洛谷是一个在线的编程题库，每个题目都有一个唯一的标识号。现在给定一个标识号'P5716'，请你编写一个Python程序，用于判断该题目号是否合法。

输入： 输入为一个字符串（题目标识号），长度为5，由大写字母'P'和数字组成。

输出： 如果给定的题目号合法，输出'YES'；否则，输出'NO'。

示例： 输入：P5716 输出：YES

输入：P573G 输出：NO

解题思路： 根据题目要求，我们需要判断给定的题目号是否合法。合法的题目号满足以下条件：

• 第一位为大写字母'P'，
• 后四位为数字。

我们可以使用Python的字符串切片功能，将给定的题目号切片成两部分。第一部分判断是否为大写字母'P'，第二部分判断是否为四位数字。

Python代码实现如下：

def is_valid_problem_id(problem_id):
    if problem_id[0] != 'P':
        return 'NO'
    if not problem_id[1:].isdigit() or len(problem_id[1:]) != 4:
        return 'NO'
    return 'YES'

problem_id = input()
print(is_valid_problem_id(problem_id))

希望以上代码能够解决你的问题。如果还有其他疑问，请随时向我提问。