牛客 NC97 - 字符串出现次数的TopK问题

字符串出现次数的TopK问题_牛客题霸_牛客网

给定一个字符串数组，再给定整数k，请返回出现次数前k名的字符串和对应的次数。

返回的答案应该按字符串出现频率由高到低排序。如果不同的字符串有相同出现频率，按字典序排序。

对于两个字符串，大小关系取决于两个字符串从左到右第一个不同字符的 ASCII 值的大小关系。

比如"ah1x"小于"ahb"，"231"<"32"。

字符仅包含数字和字母。

题解：

定义一个结构体ss，结构体中包含一个字符串str和该字符串出现的次数cnt。并定义该结构体的数组s。

首先对给定的字符串数组进行排序。

遍历排序后的字符串数组，若当前字符串与前一字符串不相同（第一次出现）则将此字符串和出现次数1存入结构体数组s中。若当前字符串与前一字符串相同则将结构体数组s中此字符串对应的cnt+1。

遍历完成后，对结构体数组s进行排序。此排序需重新自定义比较函数，按字符串的出现次数降序排序，当字符串a与b出现次数相同时按字典序排序。

排序后得到最终结果。

#include "bits/stdc++.h"
class Solution {
public:
    /**
     * return topK string
     * @param strings string字符串vector strings
     * @param k int整型 the k
     * @return string字符串vector<vector<>>
     */
    struct ss{
        string str;
        int cnt;
    };
    
    static bool cmp(ss a, ss b){
        if(a.cnt == b.cnt){
            return a.str < b.str;
        }
        else return a.cnt > b.cnt;
    }
    
    vector<vector<string> > topKstrings(vector<string>& strings, int k) {
        // write code here
        vector<vector<string>> res;
        sort(strings.begin(), strings.end());
        ss s[strings.size()];
        int j = 0;
        s[0].str = strings[0];
        s[0].cnt = 1;
        for(int i = 1 ; i < strings.size() ; i++){
            if(strings[i] != s[j].str){
                j++;
                s[j].str = strings[i];
                s[j].cnt = 1;
            }
            else s[j].cnt++;
        }
        sort(s, s+j+1, cmp);
        for(int i = 0 ; i < k ; i++){
            vector<string> v;
            v.push_back(s[i].str);
            v.push_back(to_string(s[i].cnt));
            res.push_back(v);
        }
        
        return res;
    }
};