Python判断洛谷题目编号是否合法 - 以P5716为例

使用Python判断洛谷题目编号是否合法（以P5716为例）

本篇文章提供了一个Python代码片段，用于判断给定的字符串是否为合法的洛谷题目编号，例如P5716。

**代码示例：**pythondef 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 = 'P5716'print(is_valid_problem_id(problem_id))

代码解释：

1. is_valid_problem_id(problem_id) 函数： - 接受一个字符串 problem_id 作为参数，表示待判断的题目编号。 - 首先检查第一个字符是否为 'P'，如果不是则返回 'NO'。 - 然后检查剩余部分是否为4位数字，如果不是则返回 'NO'。 - 如果上述条件都满足，则返回 'YES'，表示该字符串是合法的洛谷题目编号。

2. 示例中，我们将 problem_id 赋值为 'P5716'，并调用 is_valid_problem_id() 函数进行判断。

3. 最终输出结果为 'YES'，说明 'P5716' 是一个合法的洛谷题目编号。

您可以将代码中的 problem_id 变量修改为其他字符串进行测试。例如，'P1234'、'p1234'、'P12345' 等。