题目
给定一个字符串,只包含大写字母,求在包含同一字母的子串中,长度第k长的子串的长度,相同字母只取最长的那个子串。
输入描述:
第一行输入一个子串(长【1,100】),只包含大写字母;
第二行为k的值;
输出描述:
输出连续出现次数第k多的字母的次数;
示例1
输入:
AAAAHHHBBCDHHHH
3
输出:
2
说明:同一个字母连续出现最多的是A和H,4次;
第二多的是H,3次(H已在4次中计算过,故忽略);第三多的是BB; 故输出2;
示例2
输入:
AABAAA
2
输出:
1
解题代码
思路:
- 基于双指针,统计连续字母出现的次数,字母为key,出现次数为value;
- 相同的字母串存入字典时,值取最大的;
- 按照连续字母出现次数降序排序;
s = input().strip()
k = int(input().strip())
stats_dict = {}
pre = cur = 0
count = 0
while cur < len(s):
if s[cur] == s[pre]:
count += 1
cur += 1
else:
# 存入字典
if s[pre] not in stats_dict: # 不在 则直接存
stats_dict[s[pre]] = count
else: # 在则存最大值
stats_dict[s[pre]] = max(stats_dict.get(s[pre]), count)
pre = cur
count = 0
# 最后的存入
if s[pre] not in stats_dict:
stats_dict[s[pre]] = count
else:
stats_dict[s[pre]] = max(stats_dict.get(s[pre]), count)
# 对字典的key 按照字母出现次数降序排序
result = list(stats_dict.items())
result.sort(key=lambda e: e[1], reverse=True)
print(result[k-1][1])