2018多校赛第一场1004 Distinct Values(优先队列+思维)

题意

要求构造一个大小为N的数组,给定M个区间,区间内的元素都互不相同。

解题

初始将1~N这N个数放入优先队列中,优先队列内值小的优先。
将这M个区间按照区间左端升序排列、左端相等则按照区间长度降序排列。一开始将数组的值都置为1,然后一个区间一个区间的填数,填数的顺序就是优先队列的顺序,填一个数弹出一个元素。填下一个区间时,判断区间右侧是否大于上一个区间的右侧,如果不大于,不用填;如果大于,将上一个区间的部分元素重新入队列。这部分元素为[pre_l,min(pre_r,l-1)].
时间复杂度为O(NlogN)

标题

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#include <iostream>
#include <queue>
#include <functional>
using namespace std;

const int maxn=1e5+7;
struct node
{
    int l,r;
    node(int l,int r):l(l),r(r){}
    node(){}
    bool operator<(const node &o)const{
        return l<o.l || l==o.l&&(r-l)>(o.r-o.l);
    }
}a[maxn];
int ans[maxn];
priority_queue<int,vector<int>,greater<int> > que;
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            a[i]=node(l,r);
        }
        while(!que.empty()) que.pop();
        for(int i=1;i<=n;i++) que.push(i),ans[i]=1;
        sort(a,a+m);
        for(int i=a[0].l;i<=a[0].r;i++)
        {
            ans[i]=que.top();
            que.pop();
        }
        int now=a[0].r;
        int pre_l=a[0].l,pre_r=a[0].r;
        for(int i=1;i<m;i++)
        {
            int l=a[i].l,r=a[i].r;
            if(r<=pre_r) continue;
            for(int j=pre_l;j<=min(pre_r,l-1);j++)
            {
                 que.push(ans[j]);
            }

            for(int j=max(l,pre_r+1);j<=r;j++)
            {
                ans[j]=que.top();
                que.pop();
            }
            pre_l=l;
            pre_r=r;
        }
        for(int i=1;i<=n;i++)
            printf("%d%s",ans[i],i==n?"\n":" ");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_37685156/article/details/81173670