华为OD-----连续字母长度

题目描述
给定一个字符串，只包含大写字母，求在包含同一字母的子串中，长度第 k 长的子串的长度，相同字母只取最长的那个子串。

输入描述
第一行有一个子串(1<长度<=100)，只包含大写字母。

第二行为 k的值

输出描述
输出连续出现次数第k多的字母的次数。

用例1
输入
AAAAHHHBBCDHHHH
3
输出
2
说明
同一字母连续出现的最多的是A和H，四次；

第二多的是H，3次，但是H已经存在4个连续的，故不考虑；

下个最长子串是BB，所以最终答案应该输出2。

用例2
输入
AABAAA
2
输出
1
说明
同一字母连续出现的最多的是A，三次；

第二多的还是A，两次，但A已经存在最大连续次数三次，故不考虑；

下个最长子串是B，所以输出1。

用例3
输入
ABC
4
输出
-1
说明
只含有3个包含同一字母的子串，小于k，输出-1

用例4
输入
ABC
2
输出
1
说明
三个子串长度均为1，所以此时k = 1，k=2，k=3这三种情况均输出1。特此说明，避免歧义。

def get_klen(S,k):
    long_dirct = {}  # 存储每个字符的最长连续出现次数
    current_char = None
    current_count = 0
    for i in range(len(S)):
        if S[i] == current_char:
            current_count += 1
        else:
            if current_char is not None:
                long_dirct[current_char] = max(long_dirct.get(current_char, 0), current_count)
            current_char = S[i]
            current_count = 1
            # 添加最后一个字符的计数
    if current_char is not None:
        long_dirct[current_char] = max(long_dirct.get(current_char, 0), current_count)

    counts = sorted(long_dirct.values(), reverse=True)
    # 如果k在有效范围内内，返回第k大的值，否则返回0
    if 0<k <= len(counts):
        return counts[k - 1]
    else:
        return -1

S = input()
k=int(input())
print(get_klen(S,k))