青训/中等：字符修复问题-＞套圈太多， 自己未能理解

中等：字符修复问题->套圈太多， 自己未能理解

问题描述

小R有一个由小写字母组成的字符串 str1，长度为 n。字符串中有些位置的字符可以通过修复操作来修改为 a-z 中的任意一个字母，这个修复操作最多可以进行 m 次。现在，小R想知道，经过最多 m 次修复操作后，字符串中由相同字符组成的最长连续子串的最大长度是多少。

注意，str2表示哪些下标可以修改，如果为1那么就可以修改，如果为0那么就不行。

测试样例

样例1：

输入：n = 5, m = 2, str1 = "abcda", str2 = "01110"
输出：3

样例2：

输入：n = 7, m = 2, str1 = "abbaccb", str2 = "1001001"
输出：4

样例3：

输入：n = 3, m = 0, str1 = "aab", str2 = "101"
输出：2

示例

def solution(n, m, str1, str2):
    # Edit your code here

    return 0


if __name__ == "__main__":
    # Add your test cases here

    print(solution(5, 2, "abcda", "01110") == 3)
    print(solution(7, 2, "abbaccb", "1001001") == 4)
    print(solution(3, 0, "aab", "101") == 2)

思路：

看不懂题。。

看懂了…要得到最长的相同的字母串，限制点为你只能改这么多，是要和不能修改的墙壁相同，还是要和能修改的内部字母相同呢？比如字符“abcda” ，能修改bcd这三个字符（题目给出的修改位），理想情况下bcd都改为a在，则得到aaaaa，5个a最长了。遗憾的是，你只能改2个数字(题目限制)，因此①cd都改为b，得到三个bbb，②或者bcd改为aad，那也是三个a，aaa，③或者bcd改为ddd，也就是修改两位，也是3个相同。

那么依照这个逻辑，拥有26个字母，你能修改固定位。如何判断最长的相同字符串？

事实上，有答案，但我还理解不了。

答案

def solution(n, m, str1, str2):
    # 最大可能的连续长度不会超过字符串长度
    ans = 1
    
    # 尝试每个字母作为目标字符
    for target in 'abcdefghijklmnopqrstuvwxyz':
        # 遍历每个起始位置
        for start in range(n):
            changes_needed = 0
            curr_length = 0
            
            # 从start开始向后检查
            for i in range(start, n):
                if str1[i] != target:  # 需要修改
                    if str2[i] == '0':  # 不能修改
                        break
                    changes_needed += 1
                    if changes_needed > m:  # 超过可修改次数
                        break
                curr_length += 1
                ans = max(ans, curr_length)
    
    return ans


if __name__ == "__main__":
    # Add your test cases here

    print(solution(5, 2, "abcda", "01110") == 3)
    print(solution(7, 2, "abbaccb", "1001001") == 4)
    print(solution(3, 0, "aab", "101") == 2)

答案思路

1. 核心思路是:
   • 枚举每个可能的目标字符(a-z)
   • 枚举每个起始位置
   • 从起始位置开始尽可能往后扩展，统计需要修改的次数
   • 如果需要修改的次数超过m或遇到不能修改的位置就停止
   • 更新最大长度
2. 具体分析样例1: “abcda”, m=2, str2=“01110”
   • 比如我们选择目标字符’b’
   • 从位置1开始(字符’b’)
   • 可以将’c’和’d’都改成’b’(因为str2中对应位置是1表示可修改)
   • 这样就得到"abbba"
   • 此时最长连续子串长度为3
3. 时间复杂度:
   • 26个字母 × n个起始位置 × 最多n个长度
   • O(26 * n * n) = O(n²)

这个解决方案会遍历所有可能的情况，找出最优解。