首页 > 代码库 > [BZOJ 3669][NOI 2014]魔法森林(Link-Cut Tree)
[BZOJ 3669][NOI 2014]魔法森林(Link-Cut Tree)
题目链接:http://www.lydsy.com:808/JudgeOnline/problem.php?id=3669
记得四个月之前的NOI同步赛,我还只会玩脚丫子。。。。
记得我当时看到这个题整个人就吓傻了,完全不知道怎么做,然后NOI同步赛就这样爆零了。。。
如今我学了LCT这个神器,再看这个题,感觉不再那么难了。
其实这个题的标准解法是SPFA,改得完全认不出来的SPFA。
orz太神了,完全没见识过这么神犇的SPFA,NOIP 2014考了SPFA,NOI 2014也考了SPFA,原来SPFA也能玩出这么多花样。
我只会用LCT暴力这题,我会暴力我自豪。然后我orz了hzwer的本题lct版本代码和bin神的lct模板,自己yy乱弄了下过了这个题。
下面是我的思路,如有错误请不吝指教。
first of all,我们要构造下这题的LCT,在这题中,图有n个点m条边,于是lct里,下标在[1,n]的结点都是图中的点,下标在(n,m+n]中的结点都是图中的边,只有边对应的结点有权值,这个权值就是对应的边的b值。
然后我们用类似kruscal的方法,将每个边按照关键字a值升序排序,然后搞个并查集,维护图的连通性。
接着,我们像搞最小生成树一样,按a值从小到大不断地加边,维护并查集,加一条边时,要同时在LCT里连接这条边的两个点u和v,具体做法是先连接u和边对应的结点,再把边对应的结点和v相连。
在这个过程中,如果出现遍历到一条边u-v时,u-v本身是联通的,那么我们不加此边,但是要看LCT中u到v路径上的最大权值是不是大于这条边的b值,若是的话,则需要断掉u到v路径上最大权值的那条边的连接,加入这条边。
最后,答案就是min(a+LCT中起点对应结点到终点对应结点路径上的最大权值之和)。
PS:忍不住吐槽下,我第一次WA掉居然是因为splay操作最后忘了上传标记,orzorz。。。。。
#include <iostream> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <algorithm> #define MAXN 150050 #define MAXE 100005 #define INF 0x3f3f3f3f using namespace std; struct edge { int u,v,a,b; }edges[MAXE]; int n,m,nCount=0,ans=INF; int f[MAXN]; int ch[MAXN][2],fa[MAXN]; int val[MAXN],maxv[MAXN]; //该点值的标记、该点对应区间的最大值对应结点的下标 bool rev[MAXN]; //翻转标记 bool isRoot[MAXN]; //根节点标记 int findSet(int x) { if(f[x]==x) return x; return f[x]=findSet(f[x]); } bool operator<(edge a,edge b) { return a.a<b.a; } //Start Of Splay void updateRev(int r) //更新结点r的rev标记 { if(!r) return; swap(ch[r][0],ch[r][1]); rev[r]^=1; } void pushdown(int r) //更新结点r的rev标记 { if(rev[r]) { updateRev(ch[r][0]); updateRev(ch[r][1]); rev[r]=0; } } void pushup(int r) //更新maxv域 { int lc=ch[r][0],rc=ch[r][1]; maxv[r]=r; if(val[maxv[lc]]>val[maxv[r]]) maxv[r]=maxv[lc]; if(val[maxv[rc]]>val[maxv[r]]) maxv[r]=maxv[rc]; } void rotate(int x) //将x旋到上面一层去 { int y=fa[x],kind=ch[y][1]==x; ch[y][kind]=ch[x][!kind]; fa[ch[y][kind]]=y; fa[x]=fa[y]; fa[y]=x; ch[x][!kind]=y; if(isRoot[y]) //y是根节点,则需要更新根节点标记 { isRoot[y]=false; isRoot[x]=true; } else ch[fa[x]][ch[fa[x]][1]==y]=x; pushup(y); } void P(int r) //维护r和它的祖先们 { if(!isRoot[r]) P(fa[r]); pushdown(r); } void splay(int r) //将r旋到根节点上去 { P(r); while(!isRoot[r]) { int f=fa[r],ff=fa[f]; //父结点、祖父结点 if(isRoot[f]) rotate(r); else if((ch[ff][1]==f)==(ch[f][1]==r)) //r、r的父亲、r的祖父三点共线 rotate(f),rotate(r); else rotate(r),rotate(r); } pushup(r); } //End Of Splay //Start Of Link-Cut Tree int access(int x) //打通x到根节点的偏爱路径 { int y=0; do { splay(x); isRoot[ch[x][1]]=true; isRoot[ch[x][1]=y]=false; pushup(x); x=fa[y=x]; }while(x); return y; } void memroot(int r) //使r成为所在树的根 { access(r); splay(r); updateRev(r); } void link(int u,int v) //link操作 { memroot(u); fa[u]=v; } void cut(int u,int v) //cut操作 { memroot(u); access(v); splay(v); fa[ch[v][0]]=fa[v]; fa[v]=0; isRoot[ch[v][0]]=true; ch[v][0]=0; pushup(v); } //End Of Link-Cut Tree int query(int x,int y) //求点x到y之间路径上的最大值 { memroot(x); access(y); splay(y); return maxv[y]; } void solve(int k) //如果边k链接的两边的点u和v本身联通,而且u到v路径上的最大的b值比边k的b值大,则用边k去代替点u到v的路径 { int u=edges[k].u,v=edges[k].v,w=edges[k].b; int t=query(u,v); if(w<val[t]) { cut(edges[t-n].u,t); cut(edges[t-n].v,t); link(u,k+n); link(v,k+n); } } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n+m;i++) isRoot[i]=true; for(int i=1;i<=n;i++) f[i]=i; for(int i=1;i<=m;i++) scanf("%d%d%d%d",&edges[i].u,&edges[i].v,&edges[i].a,&edges[i].b); sort(edges+1,edges+m+1); for(int i=1;i<=m;i++) { val[n+i]=edges[i].b; maxv[n+i]=n+i; } for(int i=1;i<=m;i++) //进行类似于kruscal的操作,一边维护连通性,一边往lct里面加边 { int u=edges[i].u,v=edges[i].v,w=edges[i].b; int rootu=findSet(u),rootv=findSet(v); if(rootu!=rootv) { f[rootu]=rootv; link(u,n+i); link(v,n+i); } else solve(i); if(findSet(1)==findSet(n)) //起点到终点之间已经联通了 ans=min(ans,val[query(1,n)]+edges[i].a); } if(ans!=INF) printf("%d\n",ans); else printf("-1"); return 0; }
[BZOJ 3669][NOI 2014]魔法森林(Link-Cut Tree)