【华为OD机试 Python】滑动窗口问题（四）之连续字母长度

滑动窗口问题是机试的重点考查内容。合理移动左右指针来控制窗口大小是解决该类问题的核心。该问题涉及统计特定对象的个数，最常用的数据结构是字典。

---

---

题目描述

给定一个字符串，只包含大写字母，求在包含同一个字母的子串中，长度第k长的子串的长度，
相同字母只取最长的个子串。
若子串中只包含同一个字母的子串数小于k，则输出-1

---

输入

第一行有一个字符串（长度小于100），只包含大写字母
第二行有一个整数，表示k的值

输出

连续出现次数第k多的字母的次数

示例1

输入

AAAAHHHBBCDHHHH
3

输出

2

说明

A和H：4次
H：3次（重复）
B：2次
C和D：1次
第2多的是2次

示例2

输入

ABCD
5

输出

-1

说明

k的值超过字符串长度

解题思路及代码

1. 若当前字母不等于上一个字母，上一个字母的连续长度为 i-left。左边界调整至当前位置
2. 需额外处理边界情况，即若最后一个字母同上一个字母相同，该字母连续长度为 i - left + 1

代码如下（示例）：

s = list(input())
length = len(s)
k = int(input())  # 第k个多的
alpha = {}
left = 0
for i in range(length):
    if s[i] not in alpha.keys():  # 先判断该字母是否存在于字典中
        alpha[s[i]] = 0
    if s[i] != s[left]:  # 再判断值是否相等
        alpha[s[left]] = max(alpha[s[left]],i-left)
        left = i
    elif i == length - 1:  # 最后一个字母和上一个字母相同，需要另外处理
        alpha[s[left]] = max(alpha[s[left]],i-left+1)
res = []
for i in alpha:
    res.append(alpha[i])
res.sort(reverse=True)
print(res[k - 1] if k < length else -1)

---

总结

以上就是本文要讲的内容，遇到需要对特定对象进行计数的问题，一般用字典进行统计。若用元组进行统计，那只能用于存储不变的值。