No.114 - LeetCode496

ShellDawn

于 2019-09-15 05:11:19 发布

阅读量82

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/100844155

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用C++优先队列（最小堆）实现元素查找的方法，通过实例展示了如何在不破坏堆结构的情况下查找比当前元素大的下一个元素。此方法适用于需要高效查找并维护有序集合的应用场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> mp;
        vector<int> ans(nums1.size(),0);
        priority_queue<int,vector<int>,greater<int>> pq;
        for(int i=0;i<nums2.size();i++){
            mp[nums2[i]] = -1;
            pq.push(nums2[i]);
            if(pq.size() == 0) continue;
            while(!pq.empty()){
                if(nums2[i] > pq.top()){
                    mp[pq.top()] = nums2[i];
                    pq.pop();
                }else break;
            }
        }
        for(int i=0;i<nums1.size();i++) ans[i] = mp[nums1[i]];
        return ans;
    }
};