如果存在$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; }