P3588 [POI2015]PUS（拓扑排序+线段树）

P3588 [POI2015]PUS

对于每个$(l,r,k)$，将$k$个位置向剩下$r-l-k+1$个位置连边，边权为$1$，这样就保证$k$个位置比剩下的大

先给所有位置填$1e9$保证最优

然后拓扑排序填数

填的数不在$[1,1e9]$内或者出现环，即为不合法

但是这样边数过多会超时

于是考虑线段树优化建图

把$n$个点建成线段树，每个节点向左右儿子连边，边权为0。

这样每次连一个区间$[l,r]$就只需要$log(r-l+1)$次

注意不合法情况要枚举完整

#include<iostream>
#include<cstdio>
using namespace std;
#define N 100005
#define M 6000005
int n,s,m,u,pos[N],h[N*],L,R,no;
int p[N*],w[N*],id[N]; bool vis[N*];
int Cnt,hd[N*],nxt[M],ed[N*],poi[M],val[M],in[N*];
void adde(int x,int y,int v){
    nxt[ed[x]]=++Cnt; hd[x]=hd[x]?hd[x]:Cnt;
    ed[x]=Cnt; poi[Cnt]=y; val[Cnt]=v; ++in[y];
}
#define mid (l+r)/2
int build(int o,int l,int r){
    w[p[o]=++u]=1e9;
    if(l==r) return id[l]=u;
    adde(p[o],build(o<<,l,mid),);
    adde(p[o],build(o<<|,mid+,r),);
    return p[o];
}
void Add(int o,int l,int r,int x1,int x2,int k){
    if(x1<=l&&r<=x2){adde(k,p[o],); return ;}
    if(x1<=mid) Add(o<<,l,mid,x1,x2,k);
    if(x2>mid) Add(o<<|,mid+,r,x1,x2,k);
}
void work(){
    int tt=;
    for(int i=;i<=u;++i) if(!in[i]) h[++R]=i;
    while(L!=R){
        if(L>=N) L=;
        int x=h[++L]; ++tt;
        if(w[x]<) no=;//填的数<1
        for(int i=hd[x];i;i=nxt[i]){
            int to=poi[i];
            if(vis[to]&&w[to]>w[x]-val[i]) no=;//填的数比已给定位置上的数值小
            w[to]=min(w[to],w[x]-val[i]);
            if((--in[to])==){
                if(R>=N) R=;
                h[++R]=to;
            }
        }
    }
    if(tt<u) no=;//图中有环
}
int main(){
    scanf("%d%d%d",&n,&s,&m);
    build(,,n);
    for(int i=,Id,v;i<=s;++i){
        scanf("%d%d",&Id,&v);
        if(v<||v>1e9) no=;//给定数不合法
        w[id[Id]]=v; vis[id[Id]]=;
    }
    for(int i=,l,r,k;i<=m;++i){
        scanf("%d%d%d",&l,&r,&k); w[++u]=1e9;//新建一个中转节点
        for(int j=;j<=k;++j)
            scanf("%d",&pos[j]),adde(id[pos[j]],u,);
        if(l<pos[]) Add(,,n,l,pos[]-,u);
        if(r>pos[k]) Add(,,n,pos[k]+,r,u);
        for(int j=;j<k;++j)
            if(pos[j]+<pos[j+])
                Add(,,n,pos[j]+,pos[j+]-,u);
    }work();
    if(no) puts("NIE");
    else{
        puts("TAK");
        for(int i=;i<=n;++i) printf("%d ",w[id[i]]);
    }return ;
}