HUAWEI机试题：连续字母长度

希望有更好的解法

题目

给定一个字符串，只包含大写字母，求在包含同一字母的子串中，长度第 k 长的子串的长度，相同字母只取最长的那个子串。
条件：如果一个字母出现多次则只保留最高次数的记录，如果几个字母出现次数相同则按原字符串的出现顺序排序

输入描述

第一行有一个子串(1<长度<=100)，只包含大写字母。
第二行为 k的值

输出描述

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

示例

输入：
AABAAA
2
输出：
1
说明：
同一字母连续出现最多的A 3次
第2多，A出现2次 但A出现过连续3次，忽略
因此第2多的是B，1次，输出1

输入：
AAAAHHHBBCDHHHH
3
输出：
2
说明：
同一字母连续出现的最多的是A和H,4次,并列第1位,但算排在1,2位
第三多的是H,3次但是H出现过了,忽略
第三多的是B,B出现了2次,输出2

Python解法

str_input = input()
k = int(input())

d = dict()

i = 0
while i < len(str_input):
    key = str_input[i]
    count = 1
    for j in range(i+1,len(str_input)):
        if str_input[j] == key:
            count += 1
        else:
            break
    if key in d:
        if d[key] < count:
            d[key] = count
    elif key not in d:
        d[key] = count
    i = i + count
lst = list(d.values())
lst.sort(reverse=True)
print(lst[k-1])