1. 程式人生 > >[bzoj2816][ZJOI2012]網絡(LCT,splay)

[bzoj2816][ZJOI2012]網絡(LCT,splay)

continue open cst 節點 link const ont eof emp

傳送門

題解

話說以前還真沒見過用LCT只維護一條鏈的……好像除了樹點塗色那題……

先看一下題目規定的兩個性質

  1. 對於任意節點連出去的邊中,相同顏色的邊不超過兩條。

  2. 圖中不存在同色的環,同色的環指相同顏色的邊構成的環。

很明顯了,同一種顏色肯定是由幾條鏈組成的(雖然我根本沒有發現)

然後又要查詢邊權和維護路徑……直接上LCT吧

然後顏色數很少啊……每一個顏色開一個LCT好了

更改權值的話在每一個LCT上splay一下

修改顏色的話在原來的LCT中cut,新的LCT中link

查詢路徑直接split出來

然後差不多就這樣了

  1 //minamoto
  2 #include<cstdio>
  3 #include<map>
  4 #include<iostream>
  5 using std::swap;
  6 using std::map;
  7 template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
  8 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
  9
char buf[1<<21],*p1=buf,*p2=buf; 10 inline int read(){ 11 #define num ch-‘0‘ 12 char ch;bool flag=0;int res; 13 while(!isdigit(ch=getc())) 14 (ch==-)&&(flag=true); 15 for(res=num;isdigit(ch=getc());res=res*10+num); 16 (flag)&&(res=-res); 17
#undef num 18 return res; 19 } 20 const int N=10005; 21 int n,m,val[N],c,k; 22 struct link_cut_tree{ 23 int fa[N],ch[N][2],rev[N],mx[N],cnt[N],s[N],top; 24 inline bool isroot(int x){return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;} 25 inline void pushup(int x){ 26 mx[x]=val[x];int l=ch[x][0],r=ch[x][1]; 27 if(l) cmax(mx[x],mx[l]); 28 if(r) cmax(mx[x],mx[r]); 29 } 30 inline void pushdown(int x){ 31 if(x&&rev[x]){ 32 swap(ch[x][0],ch[x][1]); 33 rev[ch[x][0]]^=1,rev[ch[x][1]]^=1; 34 rev[x]=0; 35 } 36 } 37 void rotate(int x){ 38 int y=fa[x],z=fa[y],d=ch[y][1]==x; 39 if(!isroot(y)) ch[z][ch[z][1]==y]=x; 40 fa[x]=z,fa[y]=x,fa[ch[x][d^1]]=y,ch[y][d]=ch[x][d^1],ch[x][d^1]=y,pushup(y); 41 } 42 void splay(int x){ 43 s[top=1]=x;for(int i=x;!isroot(i);i=fa[i]) s[++top]=fa[i]; 44 while(top) pushdown(s[top--]); 45 for(int y=fa[x],z=fa[y];!isroot(x);y=fa[x],z=fa[y]){ 46 if(!isroot(y)) 47 ((ch[y][1]==x)^(ch[z][1]==y))?rotate(x):rotate(y); 48 rotate(x); 49 } 50 pushup(x); 51 } 52 void access(int x){ 53 for(int y=0;x;x=fa[y=x]) 54 splay(x),ch[x][1]=y,pushup(x); 55 } 56 void makeroot(int x){ 57 access(x),splay(x),rev[x]^=1; 58 } 59 void split(int x,int y){ 60 makeroot(x),access(y),splay(y); 61 } 62 inline void link(int x,int y){ 63 ++cnt[x],++cnt[y],makeroot(x),fa[x]=y,splay(x); 64 } 65 inline void cut(int x,int y){ 66 --cnt[x],--cnt[y],split(x,y),fa[x]=ch[y][0]=0,pushup(y); 67 } 68 int findroot(int x){ 69 access(x),splay(x),pushdown(x); 70 while(ch[x][0]) pushdown(x=ch[x][0]); 71 return x; 72 } 73 inline int query(int x,int y){ 74 split(x,y);return mx[y]; 75 } 76 }lct[15]; 77 struct edge{ 78 int u,v; 79 inline bool operator <(const edge &b)const 80 {return u<b.u||(u==b.u&&v<b.v);} 81 }; 82 map<edge,int> mp; 83 int main(){ 84 //freopen("testdata.in","r",stdin); 85 n=read(),m=read(),c=read(),k=read(); 86 for(int i=1;i<=n;++i) val[i]=read(); 87 for(int i=1;i<=m;++i){ 88 int u=read(),v=read(),w=read(); 89 edge e1=(edge){u,v},e2=(edge){v,u}; 90 mp[e1]=mp[e2]=w; 91 lct[w].link(u,v); 92 } 93 while(k--){ 94 int opt=read(); 95 switch(opt){ 96 case 0:{ 97 int x=read(),w=read(); 98 val[x]=w; 99 for(int i=0;i<c;++i) lct[i].splay(x); 100 break; 101 } 102 case 1:{ 103 int u=read(),v=read(),w=read(); 104 edge a=(edge){u,v},b=(edge){v,u}; 105 if(!mp.count(a)){puts("No such edge.");continue;} 106 int x=mp[a]; 107 if(x==w){puts("Success.");continue;} 108 if(lct[w].cnt[u]>=2||lct[w].cnt[v]>=2){puts("Error 1.");continue;} 109 if(lct[w].findroot(u)==lct[w].findroot(v)){puts("Error 2.");continue;} 110 puts("Success."); 111 lct[x].cut(u,v),lct[w].link(u,v); 112 mp[a]=mp[b]=w; 113 break; 114 } 115 case 2:{ 116 int w=read(),u=read(),v=read(); 117 if(lct[w].findroot(u)!=lct[w].findroot(v)){puts("-1");continue;} 118 printf("%d\n",lct[w].query(u,v)); 119 break; 120 } 121 } 122 } 123 return 0; 124 }

[bzoj2816][ZJOI2012]網絡(LCT,splay)