BZOJ2080 : [Poi2010]Railway

Claris發表於2016-10-07

如果存在$k$使得$i<j<k$,且$a[k]<a[i]<a[j]$,那麼$i$和$j$不能在一個棧中。

設$b[i]=\min(a[i..n])$,如果$b[j]<a[i]<a[j]$,那麼$i$和$j$不能在一個棧中。

設$c[i]$表示最大的$j$,滿足$b[j]<i$,則$i$要向位置在$[i+1,c[a[i]]]$之間所有$a[j]>a[i]$的$j$連邊,還要向$a$在$[b[i]+1,a[i]-1]$裡所有位置$<i$的$j$連邊。

然後二分圖染色即可,最後檢查是否合法。

在染色的時候,維護兩棵線段樹,第一棵按位置維護$a$的最大值,第二棵按$a$維護位置的最小值,即可不重不漏地取出所有沒有染過色的點,每次取出一個點就要將它從兩棵樹中刪除。

時間複雜度$O(n\log n)$。

 

#include<cstdio>
const int N=100010,M=262150;
int n,i,j,a[N],b[N],c[N],f[N],va[M],vb[M],pos[N],tmp,col[N],bit[3][N];
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline int min(int a,int b){return a<b?a:b;}
inline int mergea(int x,int y){
  if(!x||!y)return x+y;
  return a[x]>a[y]?x:y;
}
inline int mergeb(int x,int y){
  if(!x||!y)return x+y;
  return x<y?x:y;
}
void build(int x,int a,int b){
  if(a==b){
    pos[a]=x;
    va[x]=a;
    vb[x]=f[a];
    return;
  }
  int mid=(a+b)>>1;
  build(x<<1,a,mid),build(x<<1|1,mid+1,b);
  va[x]=mergea(va[x<<1],va[x<<1|1]);
  vb[x]=mergeb(vb[x<<1],vb[x<<1|1]);
}
void aska(int x,int a,int b,int c,int d){
  if(c<=a&&b<=d){tmp=mergea(tmp,va[x]);return;}
  int mid=(a+b)>>1;
  if(c<=mid)aska(x<<1,a,mid,c,d);
  if(d>mid)aska(x<<1|1,mid+1,b,c,d);
}
void askb(int x,int a,int b,int c,int d){
  if(c<=a&&b<=d){tmp=mergeb(tmp,vb[x]);return;}
  int mid=(a+b)>>1;
  if(c<=mid)askb(x<<1,a,mid,c,d);
  if(d>mid)askb(x<<1|1,mid+1,b,c,d);
}
inline void del(int x){
  int y=pos[x];
  for(va[y]=0,y>>=1;y;y>>=1)va[y]=mergea(va[y<<1],va[y<<1|1]);
  y=pos[a[x]];
  for(vb[y]=0,y>>=1;y;y>>=1)vb[y]=mergeb(vb[y<<1],vb[y<<1|1]);
}
void dfs(int x,int y){
  del(x);
  col[x]=y;
  int l=i+1,r=c[a[x]];
  if(l<=r)while(1){
    tmp=0;
    aska(1,1,n,l,r);
    if(!tmp)break;
    if(a[tmp]<a[x])break;
    dfs(tmp,3-y);
  }
  l=b[x]+1,a[x]-1;
  if(l<=r)while(1){
    tmp=0;
    askb(1,1,n,l,r);
    if(!tmp)break;
    if(tmp>x)break;
    dfs(tmp,3-y);
  }
}
inline void add(int*b,int x){for(;x<=n;x+=x&-x)b[x]++;}
inline int ask(int*b,int x){int t=0;for(;x;x-=x&-x)t+=b[x];return t;}
int main(){
  read(n);
  for(i=1;i<=n;i++)read(a[i]),f[a[i]]=i;
  for(b[n]=a[n],i=n-1;i;i--)b[i]=min(a[i],b[i+1]);
  for(i=1;i<=n;c[i++]=j)while(j<n&&b[j+1]<i)j++;
  build(1,1,n);
  for(i=1;i<=n;i++)if(!col[i])dfs(i,1);
  for(i=1;i<=n;i++){
    if(ask(bit[col[i]],a[i]-1)>ask(bit[col[i]],b[i]))return puts("NIE"),0;
    add(bit[col[i]],a[i]);
  }
  for(puts("TAK"),i=1;i<=n;i++)printf("%d%c",col[i],i<n?' ':'\n');
  return 0;
}