leetcode:前 K 个高频元素 && leetcode：字符串解码

前 K 个高频元素:题目描述

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入: nums = [1], k = 1
输出: [1]

说明：

你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

解题思路

先统计
然后根据统计数排序
然后取前k个元素

class Solution {
private:
    static bool cmp(vector<int> a, vector<int> b){
        return a[1] > b[1];
    }
        
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        if(nums.size() == 0 || k <= 0) return {};
        //统计
        map<int,int> mp;
        for(auto i : nums)
            mp[i]++;
        
        vector<vector<int>> v;
        for(auto it : mp)
            v.push_back({it.first,it.second});
        
        //排序
        sort(v.begin(),v.end(),cmp);
        
        //取前k个
        vector<int> res;
        for(int i = 0; i < v.size() && i < k; i++)
            res.push_back(v[i][0]);
        
        return res;
    }
};

字符串解码：题目描述

给定一个经过编码的字符串，返回它解码后的字符串。

编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。

你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。

此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。

示例:

s = “3[a]2[bc]”, 返回 “aaabcbc”.
s = “3[a2[c]]”, 返回 “accaccacc”.
s = “2[abc]3[cd]ef”, 返回 “abcabccdcdcdef”.

解题思路

比如 ab3[n2[d]]10[p]这样一个字符串
分为2部分

• ab：第一次出现数字之前的res1，数字为n
• n2[d] ：数字之后中括号中的字符串str2
• 10[p]：第一个中括号之后的字符串str3
• 对于str2和str3，需要递归调用解码的函数，得到结果为res2，res3
• 结果为：res1 + n * res2 + res3

class Solution {
public:
    string decodeString(string s) {
        if(s.size() == 0) return "";
        
        int i = s.find_first_of('[');
        if(i == -1) return s;

        //找到匹配的]的位置
        int j = i + 1;
        int stack = 1;
        while(stack > 0){
            if(s[j] == '[') stack++;
            if(s[j] == ']') stack--;
            j++;
        }
        //计算[]前面的数字
        int n = 0;
        int start = i - 1;
        int k = 0;
        while(start >= 0 && s[start] >= '0' && s[start] <= '9')
            n = pow(10,k++) * (s[start--] - '0') + n;

        //截取数字前面的字符串res1
         string res = s.substr(0,start + 1);

		// res2
        string temp = decodeString(s.substr(i + 1,j - i - 2));

        while(n-- > 0) res += temp;
        if(j != s.size()){
        	//res3
            res += decodeString(s.substr(j));
        }
        return res;
    }
};