最小替换子串长度
问题描述
输入一个长度为 4 倍数的字符串，只有A、S、D、F四个字母构成，现要求替换其中一个子串，调整为词频一样的字符串。例如ADDF，只需要替换D到S，就可以得到四个字母词频一样的字符串ASDF。求满足要求的最小子串长度。
输入样例
输入"ADDF"
输出“1”
询问豆包获得提示
根据他的思路进行代码撰写
def solution(input):
n = len(input)
target = n // 4 # 目标频次
# 统计频次
freq = {'A': 0, 'S': 0, 'D': 0, 'F': 0}
for c in input:
freq[c] += 1
# 已经平衡
if all(f == target for f in freq.values()):
return 0
# 尝试每个可能的长度
for length in range(1, n + 1):
# 检查每个起始位置
for start in range(n - length + 1):
# 计算替换该子串后的频次
temp = freq.copy()
for i in range(start, start + length):
temp[input[i]] -= 1
# 检查是否可以通过添加length个字符达到平衡
max_count = max(temp.values())
min_count = min(temp.values())
if max_count - min_count <= length:
# 检查是否可以分配length个字符来达到平衡
needed = sum(max(0, target - count) for count in temp.values())
if needed <= length:
return length
return n
