华为OD机试题-连续字母长度

题目描述

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

输入描述

第一行有一个子串（1<长度<=100）, 只包含大写字母

第二行为k的值

输出描述

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

实现脚本

import re

s = input('请输入长度1-100之间大写字母字符串：')
k = int(input('请输入次数排行：'))
r = re.search(r'^[A-Z]{1,100}$', s)
if r is None:
    print("请输入正确字符串...")
res_dic = dict()
dicts = dict()
tmp = []
dicts[s[0]] = 1
tmp.append(s[0])
temp = dict() # 用于最终的查询
temp[s[0]] = 1
for i in range(1,len(s)):
    if i == len(s)-1:
        if res_dic.get(s[i],0) <= dicts.get(s[i],0):
            res_dic[s[i ]] = dicts[s[i]]
            temp[s[i]] = temp.get(s[i],0) + 1
    if s[i] not in dicts:
        if res_dic.get(s[i-1],0) <= dicts.get(s[i-1],0):
            res_dic[s[i - 1]] = dicts[s[i - 1]]
            dicts.clear()
            dicts[s[i]] = 1
        if s[i-1] in temp:
            temp[s[i-1]] += 1
        else:
            temp[s[i]] = 1
    elif s[i] in dicts:
        if s[i] == tmp[i-1]:
            dicts[s[i]] += 1

    tmp.append(s[i])


ll = sorted(res_dic.items(),key=lambda x : x[1],reverse=True)
if k > len(ll):
    print(-1)
    exit(2)
print(ll[k-1][0],temp.get(ll[k-1][0],-1))
print(temp)
print(res_dic)

输出结果

# 正确情况
请输入长度1-100之间大写字母字符串：AAAABBCCACA
请输入次数排行：1
A 3
{'A': 3, 'C': 2}
{'A': 4, 'B': 2, 'C': 2}

# 异常情况
请输入长度1-100之间大写字母字符串：AAACCA
请输入次数排行：3
-1