Bootstrap

OD B卷【连续字母长度】

题目

给定一个字符串,只包含大写字母,求在包含同一字母的子串中,长度第k长的子串的长度,相同字母只取最长的那个子串。
输入描述:
第一行输入一个子串(长【1,100】),只包含大写字母;
第二行为k的值;
输出描述:
输出连续出现次数第k多的字母的次数;

示例1
输入:
AAAAHHHBBCDHHHH
3
输出:
2
说明:同一个字母连续出现最多的是A和H,4次;
第二多的是H,3次(H已在4次中计算过,故忽略);第三多的是BB; 故输出2;

示例2
输入:
AABAAA
2
输出:
1
 

解题代码

思路:

  1. 基于双指针,统计连续字母出现的次数,字母为key,出现次数为value;
  2. 相同的字母串存入字典时,值取最大的;
  3. 按照连续字母出现次数降序排序;

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])

;