python 練習題 NC97 字符串出現次數的TopK問題

地址：https://www.nowcoder.com/practice/fd711bdfa0e840b381d7e1b82183b3ee?tpId=188&&tqId=38637&rp=1&ru=/activity/oj&qru=/ta/job-code-high-week/question-ranking

 

 

'''
描述
給定一個字符串數組，再給定整數k，請返回出現次數前k名的字符串和對應的次數。
返回的答案應該按字符串出現頻率由高到低排序。如果不同的字符串有相同出現頻率，按字典序排序。
對於兩個字符串，大小關系取決於兩個字符串從左到右第一個不同字符的 ASCII 值的大小關系。
比如"ah1x"小於"ahb"，"231"<”32“
字符僅包含數字和字母

[要求]
如果字符串數組長度為N，時間復雜度請達到O(N \log K)O(NlogK)

示例1
輸入：
["a","b","c","b"],2
復制
返回值：
[["b","2"],["a","1"]]
說明：
"b"出現了2次，記["b","2"]，"a"與"c"各出現1次，但是a字典序在c前面，記["a","1"]，最后返回[["b","2"],["a","1"]]
 
示例2
輸入：
["123","123","231","32"],2
返回值：
[["123","2"],["231","1"]]
說明：
 "123"出現了2次，記["123","2"]，"231"與"32"各出現1次，但是"231"字典序在"32"前面，記["231","1"]，最后返回[["123","2"],["231","1"]] 

'''

'''
1.先計數並存轉成list格式
2.先按照 出現數字降序 再按照字符升序排序
'''

#
# return topK string
# @param strings string字符串一維數組 strings
# @param k int整型 the k
# @return string字符串二維數組
#
class Solution:
    def topKstrings(self , strings , k ):
        # write code here
        res = {}
        for i in strings:
            if i not in res.keys():
                res[i] = 1
            else:
                res[i] +=1
        resList = []
        for i in res.keys():
            resList.append([i,res[i]])
        resList.sort(key=lambda i : (-i[1],i[0]))
        return resList[:k]