代码随想录训练营第11天|类型推导/操作符重载

jiyisuifeng1991

已于 2024-09-04 14:00:01 修改

阅读量190

点赞数 1

文章标签：算法 leetcode 数据结构

于 2024-08-24 20:32:24 首次发布

本文链接：https://blog.csdn.net/jiyisuifeng1991/article/details/141503532

版权

150. 逆波兰表达式求值

class Solution {
public:
    int evalRPN(vector<string>& tokens) {
        unordered_set<string> operators{"+","-","*","/"};
        stack<int> st;
        for(auto token:tokens){
            if(operators.count(token)){
                int num1=st.top();
                st.pop();
                int num2=st.top();
                st.pop();
                if(token=="+")
                    st.push(num2+num1);
                else if(token=="-")
                    st.push(num2-num1);
                else if(token=="*")
                    st.push(num2*num1);
                else
                    st.push(num2/num1);
                
            }
            else{
                st.push(stoi(token));
            }

        }
        return st.top();
    }
};

239. 滑动窗口最大值

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        vector<int> res;
        deque<int> q;
        for(int i=0;i<nums.size();i++){
            while(!q.empty()&q.front()<i+1-k)
                q.pop_front();
            while(!q.empty()&&nums[q.back()]<nums[i])
                q.pop_back();
            q.push_back(i);
            if(i>=k-1)
                res.push_back(nums[q.front()]);            
        }
        return res;
    }
};

第一个while调整队首，数据不能太老，处理至i位，窗口起点i+1-k，比起点还小的无效；

第二个while调整队尾，数据不能太小，因为求窗口内最大值，所以凡是小于nums[i]的均无用，pop完毕后，实际就形成了单调递减队列，队首表征当前窗口最大值的索引。

347.前 K 个高频元素

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> count;
        for(auto num:nums){
            count[num]++;
        }
        auto cmp=[](pair<int,int> a, pair<int,int> b){return a.second>b.second;};
        priority_queue<pair<int,int>, vector<pair<int,int>>, decltype(cmp)> pq(cmp);
        for(auto [key,val]:count){
            pq.push({key,val});
            if(pq.size()>k)
                pq.pop();
        }
        vector<int> res;
        while(!pq.empty()){
            res.push_back(pq.top().first);
            pq.pop();
        }
        return res;
    }
};

使用decltype自动推导cmp的类型，或者显式写出：function<bool(pair<int,int>,pair<int,int>)>