BZOJ4699 : 树上的最短路
这道题主要是要解决以下两个问题:
问题1:
给定一个点$x$,如何取出所有经过它的下水道?
一条下水道经过$x$等价于它起点在$x$的子树里面且终点不在$x$的子树里面,或者两端点的lca就是$x$。
对于第一种情况,也就是说起点在$x$的dfs序子区间里,终点小于$st[x]$或者大于$en[x]$。
假设下水道是$A-B$向$C-D$连边,并且$st[A]<st[B]$,
那么只需要不断查询$st[A]$在$[L,R]$里$st[B]$最大的下水道,以及$st[B]$在$[L,R]$里$st[A]$最小的下水道即可。
用线段树按dfs序维护终点$dfn$最小和最大的下水道,然后不断取出即可。
注意到Dijkstra的时候,$dis$不降,所以对于一条下水道,只有$A-B$里第一个被访问到的点才会用到它,所以在使用完毕后直接删除即可。
问题2:
如何求最短路?
考虑转化问题,改成求起点到每条有向边的最短路,这个最短路包括这条边本身的权值。
按距离维护一个小根堆,里面有两种元素:
$1.$某条有向树边$x->y$
$2.$某条下水道$A-B,C-D$
每次取出堆顶元素,如果是树边,那么枚举$y$的出边,同时将经过$y$的下水道都取出即可。
如果是下水道,那么暴力将$C-D$路径上所有没有访问过的点的出边都取出。
因为每个点只需要取一次,所以用并查集完成路径压缩即可。
总时间复杂度$O((n+m)\log n)$,空间复杂度$O(n+m)$。
#include<cstdio> #include<algorithm> using namespace std; typedef long long ll; typedef pair<ll,int>P; const int N=250010,M=100010,T=262150,BUF=12000000,OUT=5000000; char Buf[BUF],*buf=Buf,Out[OUT],*ou=Out;int Outn[30],Outcnt; int n,m,S,i,j,C,D,x,y,z,g[N],v[N<<1],w[N<<1],nxt[N<<1],ed,fa[N],G[N],NXT[M];bool vis[M]; int size[N],f[N],d[N],son[N],top[N],dfn,st[N],en[N]; int l1[N],r1[N],l2[N],r2[N],ea[M],eb[M],va[T],vb[T],cur,q[99],pos[M],l; ll ans[N];P t,h[N*2+M]; struct E{int u,v,z,a,b,c;}e[M]; void sorta(int l,int r){ int i=l,j=r,m=e[ea[(l+r)>>1]].a; do{ while(e[ea[i]].a<m)i++; while(e[ea[j]].a>m)j--; if(i<=j)swap(ea[i],ea[j]),i++,j--; }while(i<=j); if(l<j)sorta(l,j); if(i<r)sorta(i,r); } void sortb(int l,int r){ int i=l,j=r,m=e[eb[(l+r)>>1]].b; do{ while(e[eb[i]].b<m)i++; while(e[eb[j]].b>m)j--; if(i<=j)swap(eb[i],eb[j]),i++,j--; }while(i<=j); if(l<j)sortb(l,j); if(i<r)sortb(i,r); } inline void read(int&a){for(a=0;*buf<48;buf++);while(*buf>47)a=a*10+*buf++-48;} inline void write(ll x){ if(!x)*ou++=48; else{ for(Outcnt=0;x;x/=10)Outn[++Outcnt]=x%10+48; while(Outcnt)*ou++=Outn[Outcnt--]; } } inline void add(int x,int y,int z){v[++ed]=y;w[ed]=z;nxt[ed]=g[x];g[x]=ed;} inline void ADD(int x,int y){NXT[y]=G[x];G[x]=y;} void dfs(int x){ size[x]=1; for(int i=g[x];i;i=nxt[i])if(v[i]!=f[x]){ f[v[i]]=x,d[v[i]]=d[x]+1; dfs(v[i]),size[x]+=size[v[i]]; if(size[v[i]]>size[son[x]])son[x]=v[i]; } } void dfs2(int x,int y){ st[x]=++dfn;top[x]=y; if(son[x])dfs2(son[x],y); for(int i=g[x];i;i=nxt[i])if(v[i]!=son[x]&&v[i]!=f[x])dfs2(v[i],v[i]); en[x]=dfn; } inline int lca(int x,int y){ for(;top[x]!=top[y];x=f[top[x]])if(d[top[x]]<d[top[y]])swap(x,y); return d[x]<d[y]?x:y; } inline int mergea(int x,int y){ if(!x||!y)return x+y; return e[ea[x]].b>e[ea[y]].b?x:y; } inline int mergeb(int x,int y){ if(!x||!y)return x+y; return e[eb[x]].a<e[eb[y]].a?x:y; } void build(int x,int a,int b){ if(a==b){ pos[a]=x; va[x]=vb[x]=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 ask(int x,int a,int b){ if(C<=a&&b<=D){q[cur++]=x;return;} int mid=(a+b)>>1; if(C<=mid)ask(x<<1,a,mid); if(D>mid)ask(x<<1|1,mid+1,b); } inline void dela(int x){ va[x=pos[x]]=0; for(x>>=1;x;x>>=1)va[x]=mergea(va[x<<1],va[x<<1|1]); } inline void delb(int x){ vb[x=pos[x]]=0; for(x>>=1;x;x>>=1)vb[x]=mergeb(vb[x<<1],vb[x<<1|1]); } inline void put(const P&x){ h[++l]=x; for(int i=l;i>1&&h[i]<h[i>>1];i>>=1)swap(h[i],h[i>>1]); } inline void get(){ t=h[1],h[1]=h[l--]; for(int i=1;;){ ll tmp=h[i].first;int j=0; if((i<<1)<=l&&h[i<<1].first<tmp)tmp=h[j=i<<1].first; if((i<<1|1)<=l&&h[i<<1|1].first<tmp)j=i<<1|1; if(j)swap(h[i],h[j]),i=j;else return; } } inline void up(int x,ll d){ if(fa[x]!=x)return; fa[x]=f[x]; ans[x]=d; for(i=g[x];i;i=nxt[i])put(P(d+w[i],-i)); for(i=G[x];i;i=NXT[i])if(!vis[i])vis[i]=1,put(P(d+e[i].c,i)); C=l1[st[x]],D=r1[en[x]]; if(C<=D){ cur=0;ask(1,1,m); while(1){ for(y=i=0;i<cur;i++)y=mergea(y,va[q[i]]); if(!y)break; if(e[ea[y]].b<=en[x])break; dela(y); if(!vis[y=ea[y]])vis[y]=1,put(P(d+e[y].c,y)); } } C=l2[st[x]],D=r2[en[x]]; if(C<=D){ cur=0;ask(1,1,m); while(1){ for(y=i=0;i<cur;i++)y=mergeb(y,vb[q[i]]); if(!y)break; if(e[eb[y]].a>=st[x])break; delb(y); if(!vis[y=eb[y]])vis[y]=1,put(P(d+e[y].c,y)); } } } int F(int x){return fa[x]==x?x:fa[x]=F(fa[x]);} int main(){ fread(Buf,1,BUF,stdin);read(n),read(m),read(S); for(i=1;i<n;i++)read(x),read(y),read(z),add(x,y,z),add(y,x,z); dfs(d[1]=1),dfs2(1,1); for(i=1;i<=m;i++){ read(e[i].u),read(e[i].v),read(x),read(y),read(e[i].c); ADD(lca(x,y),i); e[i].z=d[lca(e[i].u,e[i].v)]; x=st[x],y=st[y]; if(x>y)swap(x,y); e[i].a=x,e[i].b=y; ea[i]=eb[i]=i; } if(!m)m=1; sorta(1,m); sortb(1,m); build(1,1,m); for(j=m+1,i=n;i;i--){ while(j>1&&e[ea[j-1]].a>=i)j--; l1[i]=j; } for(j=0,i=1;i<=n;i++){ while(j<m&&e[ea[j+1]].a<=i)j++; r1[i]=j; } for(j=m+1,i=n;i;i--){ while(j>1&&e[eb[j-1]].b>=i)j--; l2[i]=j; } for(j=0,i=1;i<=n;i++){ while(j<m&&e[eb[j+1]].b<=i)j++; r2[i]=j; } for(i=1;i<=n;i++)fa[i]=i; up(S,0); while(l){ get(); if(t.second>0){ z=e[t.second].z; for(x=F(e[t.second].u);d[x]>=z;x=F(x))up(x,t.first); for(x=F(e[t.second].v);d[x]>=z;x=F(x))up(x,t.first); }else up(v[-t.second],t.first); } for(i=1;i<=n;i++)write(ans[i]),*ou++='\n'; fwrite(Out,1,ou-Out,stdout); return 0; }