atcoder ABC214 E Packing Under Range Regulations

atcoder ABC214 E Packing Under Range Regulations

传送门

题目大意

有$10^9$个盒子，有$n$个球需要放入，第$i$个只可以放入$[l_i,r_i]$区间的盒子内，问是否所以盒子可以放下?

思路

小贪心

不难想到$i$位置可以被多个球放时，应该尊老爱幼让给$r$最小的球。

所以我们可以暴力$i$的盒子位置，用一个小根优先队列放入$r(l\le i)$排序，每次取队头的球放入盒子，如果$i<r$时那么肯定不可以放完（此处建议用样例模拟说明）。

当$i<r$时，$r$早已入队，证明在$r$前面的 盒子里 的球有更小的$r$，且已经占用了整个区间。

样例输入

2
3
1 2
2 3
3 3
5
1 2
2 3
3 3
1 3
999999999 1000000000

样例输出

Yes
No

但是我们现在有一个问题：盒子编号到了$10^9$，暴力定会换来TLE。

我们可以发现，各个球的区间之间距离很大，所以我们在当前的$i$没有球的时候，直接跳到下一个球的$l$就好了。

CODE

#include<bits/stdc++.h>
using namespace std;

const int maxn=2e5+5;

int n,t;

priority_queue<int,vector<int>,greater<int>>q;//小根堆

map<int,vector<int>>mp;//这里有点像milt_map，是map的一个值对应一个数组，数组里存储值

set<int>ball;

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        mp.clear();
        while (!q.empty())
        {
            q.pop();
        }
        ball.clear();//清空各类数组、队列
        
        int bj=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            mp[x].push_back(y);//加入map
            ball.insert(x);//将l加入有序集合
        }
        ball.insert(1e9+1);//这里要注意，最后一个集合值应为无限大，这样才可以使while退出
        int now=0;//相当与上文的i
        while(now<=1e9)
        {
            if(mp.count(now))
            {
                for(int i=0;i<mp[now].size();i++)//将l=i的球的r加入优先队列
                {
                    q.push(mp[now][i]);
                }
            }
            if(q.empty()) now=*ball.lower_bound(now);//队列空时直接跳入下一个球的l
            else
            {
                int x=q.top();
                q.pop();
                if(x<now)//如果r已经小于i证明为时以晚
                {
                    printf("No\n");
                    bj=1;
                    break;
                }
                now++;//正常放入
            }
        }
        if(q.size()&&!bj)
        {
            printf("No\n");
        }
        else if(!bj)
        {
            printf("Yes\n");
        }
    }
}

注

代码中采用set(有序的集合)来将$l$排序，当优先队列中没有数的时候直接lower_bround（二分查找）下一个$l$就好了。