题437.Leetcode-Q6390. 对顶堆/哈希-滑动子数组的美丽值（区间第x小数？）

---

题437.Leetcode-Q6390. 对顶堆/哈希-滑动子数组的美丽值

---

一、题目

二、题解

要求寻找区间第k小数，共有以下三种做法，本题可用第二种，第三种通过。

法1：维护一个单调不减的数组，插入时二分O(logn)时间找到位置，O(n)时间插入，总时间为O(nlogn)，删除同理；O(1)时间查找第x小数。记得卡常（虽然本题卡了也过不去）。代码如下：

//单调数组，卡常，超时
#define pb push_back
#define ins(x) a.insert(upper_bound(a.begin(),a.end(),x),x)
#define que(x) a[x-1]
#define del(x) a.erase(lower_bound(a.begin(),a.end(),x))

class Solution {

vector<int> a;

public:
    vector<int> getSubarrayBeauty(vector<int>& nums, int k, int x) {
        vector<int> res;
        int i=0,l=0;
        for(i=0;i<k;i++) ins(nums[i]);
        res.pb(min(que(x),0));
        for(;i<nums.size();i++)
        {
            del(nums[l++]);
            ins(nums[i]);
            res.pb(min(que(x),0));
        }
        return res;
    }
};

法2：用两个set构造一个对顶堆，第一个set用于存放当前区间最小的x个数，第二个set用于存放当前区间剩下的数。插入删除O(logn)，查找第x小数O(1)。代码如下：

//双set对顶端，正好够过
class Solution {

multiset<int> s1,s2;

public:
    void ins(int x,int rk)
    {
        int a=-55,b=55;
        if(!s1.empty()) a=*s1.rbegin();
        if(!s2.empty()) b=*s2.begin();
        if(s1.size()<rk)
        {
            if(x<=b) s1.insert(x);
            else
            {
                s1.insert(b);
                auto it=s2.find(b);
                if(it!=s2.end()) s2.erase(it);
                s2.insert(x);
            }
        }
        else
        {
            if(x<=a)
            {
                s1.insert(x);
                auto it=s1.find(a);
                if(it!=s2.end()) s1.erase(it);
                s2.insert(a);
            }
            else s2.insert(x);
        }
    }

    void del(int x)
    {
        auto it1=s1.find(x),it2=s2.find(x);
        if(it1!=s1.end()) s1.erase(it1);
        else if(it2!=s2.end()) s2.erase(it2);
    }

    int que(int x)
    {
        return *s1.rbegin();
    }

    vector<int> getSubarrayBeauty(vector<int>& nums, int k, int x) {
        vector<int> tmp,res;
        int i=0,l=0;

        for(i=0;i<k;i++) tmp.push_back(nums[i]);
        sort(tmp.begin(),tmp.end());
        for(int j=0;j<x;j++) s1.insert(tmp[j]);
        for(int j=x;j<k;j++) s2.insert(tmp[j]);

        res.push_back(min(que(x),0));
        for(;i<nums.size();i++)
        {
            del(nums[l++]);
            ins(nums[i],x);
            res.push_back(min(que(x),0));
        }
        return res;
    }
};

法3：针对这里数组中数的大小为-50~50，很小，所以可以开一个递增map去存当前区间的各个数的个数，然后查找的时候遍历map（数范围很小，所以不慢），统计当前数的个数cnt和x比较，大于等于时对应数即为第x小数。代码如下：

class Solution {

map<int,int> m;

public:
    int que(int x)
    {
        int cnt=0,ans;
        for(auto it=m.begin();it!=m.end();it++)
        {
            cnt+=it->second;
            if(cnt>=x)
            {
                ans=it->first;
                break;
            }
        }
        return ans;
    }

    vector<int> getSubarrayBeauty(vector<int>& nums, int k, int x) {
        vector<int> res;
        int i,l=0;
        for(i=0;i<k;i++) m[nums[i]]++;
        res.push_back(min(que(x),0));
        for(;i<nums.size();i++)
        {
            m[nums[l++]]--;
            m[nums[i]]++;
            res.push_back(min(que(x),0));
        }
        return res;
    }
};

---