ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

CF1158C Permutation recovery(线段树优化建图)

2021-03-05 21:35:19  阅读:216  来源: 互联网

标签:10 recovery idx int tr 建图 CF1158C build


如果直接拓扑建图,那么建图复杂度太高,因此我们考虑线段树优化建图

也就是区间连边,这样也能达到对应的目的,因为是单向边,建一个树即可

之后跑一下拓扑排序

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int N=1e6+10;
const int mod=998244353;
struct node{
    int l,r;
}tr[N*4];
int ans[N],nxt[N];
int id[N],in[N*10],d[N*10];
int h[N*10],ne[N*10],e[N*10],idx;
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
    in[b]++;
}
void build(int u,int l,int r){
    h[u]=-1,in[u]=d[u]=0;
    if(l==r){
        tr[u]={l,r};
        id[l]=u,d[u]=l;
    }
    else{
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        add(u<<1,u),add(u<<1|1,u);
    }
}
void modify(int u,int l,int r,int x){
    if(tr[u].l>=l&&tr[u].r<=r){
        add(u,x);
        return ;
    }
    int mid=tr[u].l+tr[u].r>>1;
    if(l<=mid){
        modify(u<<1,l,r,x);
    }
    if(r>mid)
        modify(u<<1|1,l,r,x);
}
int main(){
    ios::sync_with_stdio(false);
    int t;
    cin>>t;
    while(t--){
        int n;
        cin>>n;
        int i;
        idx=0;
        for(i=1;i<=n;i++){
            cin>>nxt[i];
        }
        build(1,1,n);
        for(i=1;i<=n;i++){
            if(nxt[i]!=-1){
                modify(1,i+1,nxt[i]-1,id[i]);
                if(nxt[i]!=n+1){
                    modify(1,i,i,id[nxt[i]]);
                }
            }
        }
        queue<int> q;
        for(i=1;i<=n;i++){
            if(!in[id[i]]){
                q.push(id[i]);
            }
        }
        int res=0;
        while(q.size()){
            int t=q.front();
            q.pop();
            if(d[t]) ans[d[t]]=++res;
            for(int i=h[t];i!=-1;i=ne[i]){
                int j=e[i];
                in[j]--;
                if(!in[j])
                    q.push(j);
            }
        }
        if(res!=n){
            cout<<-1<<endl;
        }
        else{
            for(i=1;i<=n;i++){
                cout<<ans[i]<<" ";
            }
            cout<<endl;
        }
    }
    return 0;
}
View Code

 

标签:10,recovery,idx,int,tr,建图,CF1158C,build
来源: https://www.cnblogs.com/ctyakwf/p/14488567.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有