第十四届蓝桥杯大赛软件赛省赛Python-研究生组-C.填充

问题描述

有一个长度为 n 的 01 串，其中有一些位置标记为 ?，这些位置上可以任意填充 0 或者 1，请问如何填充这些位置使得这个 01 串中出现互不重叠的 00 和 11 子串最多，输出子串个数。

输入格式
输入一行包含一个字符串。

输出格式
输出一行包含一个整数表示答案。

样例输入
1110?0

样例输出
2

样例说明
如果在问号处填 0 ，则最多出现一个 00 和一个 11：111000。

评测用例规模与约定
对于所有评测用例，1≤n≤$10^6$。

---

题解

考虑使用动态规划解决问题。定义 $f_i$表示只考虑 S 的前 i 个字符时可凑出的最多 00 或 11 子串。

首先，对于状态初始化有：$f_0=f_1=0$，因为长度不足 2 所以一定无法凑出 00 或 11。

接下来考虑状态转移方程，当 i≥2 时我们讨论 $S_i$ 和 $S_{i-1}$​ 的关系：
若 $S_i$与 $S_{i-1}$ 其中一个等于 ? 或者 $S_i=S_{i-1}$​，此时$S_i$和 $S_{i-1}$ 可以凑成一对，即可以得到转移方程：
$f_i=f_{i-2}+1$

若 $S_i$与 $S_{i-1}$​ 的关系不满足上述条件，则说明无法凑成一对，即有转移方程：
$f_i=f_{i-1}$

最终答案即为 $f_n$​，时间复杂度：O(n)。

代码

import os
import sys

# 请在此输入您的代码

s = input()
n = len(s)
s = '?' + s
f = [0] * (n + 1)
for i in range(2, n + 1):
    if s[i - 1] == '?' or s[i] == '?' or (s[i] == s[i - 1]):
        f[i] = f[i - 2] + 1
    else:
        f[i] = f[i - 1]
print(f[n])