BZOJ2816-[ZJOI2012]网络
BZOJ2816-[ZJOI2012]网络
题意:
题解:
丧心病狂数据结构题,题解想一会就想出来了,但打了半天.
可以发现每一种颜色都是由几条链构成,然后就可以用splay维护每种颜色的每条链,然后连接的时候如果是连接两条链的头部或两条链的尾部,需要翻转其中一条链.所以还要打一个区间翻转的标记.
#include<cstdio> #include<algorithm> #include<stack> #include<map> using namespace std; const int maxn=100000; struct splay_tree{ struct node{ int ch[2],fa,val,tot,sz; bool rev; }a[maxn+10]; int get_child(int p){ if(a[p].fa==-1) return -1; else if(a[a[p].fa].ch[0]==p) return 0; else if(a[a[p].fa].ch[1]==p) return 1; else return -1; } void push_down(int p){ if(a[p].rev){ if(a[p].ch[0]!=-1) a[a[p].ch[0]].rev^=1; if(a[p].ch[1]!=-1) a[a[p].ch[1]].rev^=1; swap(a[p].ch[0],a[p].ch[1]); a[p].rev=0; } } void push_fa_tag(int p,int fa){ stack<int> S; for(int i=p;i!=fa;i=a[i].fa) S.push(i); for(;!S.empty();S.pop()) push_down(S.top()); } void update(int p){ a[p].tot=a[p].val; a[p].sz=1; for(int i=0;i<=1;++i) if(a[p].ch[i]!=-1){ a[p].tot=max(a[p].tot,a[a[p].ch[i]].tot); a[p].sz+=a[a[p].ch[i]].sz; } } void rotate(int p){ int f=a[p].fa,d=get_child(p),dd=get_child(f); a[f].ch[d]=a[p].ch[d^1]; if(a[f].ch[d]!=-1) a[a[f].ch[d]].fa=f; a[p].ch[d^1]=f; a[p].fa=a[f].fa; if(dd!=-1) a[a[p].fa].ch[dd]=p; a[f].fa=p; update(f); update(p); } void splay(int p,int fa){ for(push_fa_tag(p,fa);a[p].fa!=fa;rotate(p)) if(a[a[p].fa].fa!=fa) if(get_child(p)==get_child(a[p].fa)) rotate(a[p].fa); else rotate(p); } int get_root(int p){ for(;a[p].fa!=-1;p=a[p].fa); return p; } }graph[10]; int n,m,c,k,v[maxn+10]; struct edge{ int u,v; edge(int u,int v):u(min(u,v)),v(max(u,v)){} bool operator<(const edge &t)const{ if(u!=t.u) return u<t.u; return v<t.v; } }; map<edge,int> color; void add_edge(int l,int r,int co){ color[edge(l,r)]=co; graph[co].splay(l,-1); graph[co].splay(r,-1); if(graph[co].a[l].ch[1]!=-1){ graph[co].a[l].rev^=1; graph[co].push_down(l); } if(graph[co].a[r].ch[0]!=-1){ graph[co].a[r].rev^=1; graph[co].push_down(r); } graph[co].a[l].ch[1]=r; graph[co].a[r].fa=l; graph[co].update(l); } void change_node(int p,int v){ for(int i=0;i<c;++i){ graph[i].splay(p,-1); graph[i].a[p].val=v; graph[i].update(p); } } void change_edge(int l,int r,int co){ edge eg(l,r); map<edge,int>::iterator it=color.find(eg); if(it==color.end()){ printf("No such edge.\n"); return; } if(co==it->second){ printf("Success.\n"); return; } graph[co].splay(l,-1); if(graph[co].a[l].ch[0]!=-1&&graph[co].a[l].ch[1]!=-1){ printf("Error 1.\n"); return; } graph[co].splay(r,-1); if(graph[co].a[r].ch[0]!=-1&&graph[co].a[r].ch[1]!=-1){ printf("Error 1.\n"); return; } if(graph[co].get_root(l)==graph[co].get_root(r)){ printf("Error 2.\n"); return; } if(graph[co].a[l].ch[1]!=-1){ graph[co].a[l].rev^=1; graph[co].push_down(l); } if(graph[co].a[r].ch[0]!=-1){ graph[co].a[r].rev^=1; graph[co].push_down(r); } graph[co].a[l].ch[1]=r; graph[co].a[r].fa=l; graph[co].update(l); int prev=it->second,d; it->second=co; graph[prev].splay(l,-1); graph[prev].splay(r,l); if(graph[prev].a[l].ch[0]==r) d=0; else if(graph[prev].a[l].ch[1]==r) d=1; graph[prev].a[l].ch[d]=-1; graph[prev].a[r].fa=-1; graph[prev].update(l); printf("Success.\n"); }; void query(int l,int r,int co){ int ans=max(graph[co].a[l].val,graph[co].a[r].val),d; if(l==r){ printf("%d\n",ans); return; } if(graph[co].get_root(l)!=graph[co].get_root(r)){ printf("-1\n"); return; } graph[co].splay(l,-1); graph[co].splay(r,l); if(graph[co].a[l].ch[0]==r) d=0; else if(graph[co].a[l].ch[1]==r) d=1; if(graph[co].a[r].ch[d^1]!=-1) ans=max(ans,graph[co].a[graph[co].a[r].ch[d^1]].tot); printf("%d\n",ans); } int main(){ scanf("%d%d%d%d",&n,&m,&c,&k); for(int i=1;i<=n;++i) scanf("%d",&v[i]); for(int i=0;i<c;++i) for(int j=1;j<=n;++j){ graph[i].a[j].ch[0]=graph[i].a[j].ch[1]=graph[i].a[j].fa=-1; graph[i].a[j].val=v[j]; graph[i].update(j); } for(int i=1;i<=m;++i){ int l,r,co; scanf("%d%d%d",&l,&r,&co); add_edge(l,r,co); } for(int i=1;i<=k;++i){ int op,l,r,co; scanf("%d",&op); if(op==0){ scanf("%d%d",&l,&r); change_node(l,r); }else if(op==1){ scanf("%d%d%d",&l,&r,&co); change_edge(l,r,co); }else if(op==2){ scanf("%d%d%d",&co,&l,&r); query(l,r,co); } } return 0; }
