【自定义】抢妹子大作战

题目链接https://www.luogu.org/problemnew/show/U25598

这个题是我昨天在补HDU5361时根据它的优化方法想出来的一道题。我写的标解是并查集。

题意

  有n个椅子n个人，每个人都有个目标椅子，但是如果目标椅子已经被人抢了，就继续往后找。n<=10^7

  输出最后每把椅子上做的是谁。

分析

  开一个并查集，pos=find(i)代表第i把椅子右边最近的空椅子。初始时所有的椅子都是空的所以p[i]=i，最近的时它本身。

  当第i把椅子被选了以后，则更新p[find(i)]=find(i+1)。

  代码如下

  

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>

using namespace std;
const int maxn=10000000+10;
int a[maxn],v[maxn],p[maxn];
int T,n;
int find(int x){
    return p[x]==x?x:p[x]=find(p[x]);
}
int main(){
   // freopen("out.txt","r",stdin);
   // freopen("out2.txt","w",stdout);
    scanf("%d",&T);
    for(int t=1;t<=T;t++){
        scanf("%d",&n);
        memset(v,0,sizeof(v));
        for(int i=0;i<=n+2;i++)p[i]=i;
        for(int i=0;i<=n;i++){
            scanf("%d",&a[i]);
        }
        bool ok=1;

        for(int i=0;i<=n;i++){
            int pos=find(a[i]);
           // cout<<i<<" "<<pos<<endl;
            if(pos>n){
                ok=0;
                break;
            }
            v[pos]=i;
            p[pos]=find(pos+1);
        }
        if(!ok){
            printf("-1");
        }
        else{
            for(int i=0;i<=n;i++)
                printf("%d ",v[i]);
        }
        printf("
");
    }
return 0;
}