一颗树n个点,n-1条边,经过每条边都要花费一定的时间,任意两个点都是联通的。
有K个人(分布在K个不同的点)要集中到一个点举行聚会。
聚会结束后需要一辆车从举行聚会的这点出发,把这K个人分别送回去。
请你回答,对于i=1~n,如果在第i个点举行聚会,司机最少需要多少时间把K个人都送回家。
标签:出发点 led input printf ons for 必须 mil lld
题解:我们先高出这k个点的虚树,先考虑出发点u这个点是虚树上的点,且必须返回出发点时答案是什么。显然答案就是这个虚树的所有边的长度之和*2。
那么如果要返回出发点呢?如果我们再v结束,那么答案就会减去dis(u,v),那么显然dis(u,v)越大越好,所以树形DP求一下每个点到虚树上点的距离最大值即可。
然后如果u不在虚树上呢?那么它还要先走到虚树上,所以再维护一下每个点到虚树上点的距离最小值即可。
代码还是很不可读的~
#include <cstdio> #include <cstring> #include <iostream> using namespace std; typedef long long ll; const int maxn=500010; ll ans; int to[maxn<<1],next[maxn<<1],val[maxn<<1],head[maxn],p[maxn],q[maxn],s[maxn],fa[maxn],Q[maxn],st[maxn]; ll dep[maxn],f[maxn][2],g[maxn],h[maxn][2],k[maxn]; int n,m,cnt,top,np,mp; inline int rd() { int ret=0,f=1; char gc=getchar(); while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘)f=-f; gc=getchar();} while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+gc-‘0‘,gc=getchar(); return ret*f; } void add(int a,int b,int c) { to[cnt]=b,val[cnt]=c,next[cnt]=head[a],head[a]=cnt++; } void dfs(int x) { p[x]=++p[0],Q[p[0]]=x; for(int i=head[x];i!=-1;i=next[i]) if(to[i]!=fa[x]) dep[to[i]]=dep[x]+val[i],fa[to[i]]=x,dfs(to[i]); q[x]=p[0]; } int main() { n=rd(),m=rd(); int i,a,b,c,x,y; ll d; memset(head,-1,sizeof(head)); for(i=1;i<n;i++) a=rd(),b=rd(),c=rd(),add(a,b,c),add(b,a,c); dfs(1); np=n,mp=0; memset(f,0xc0,sizeof(f)),memset(g,0xc0,sizeof(g)),memset(h,0x3f,sizeof(h)),memset(k,0x3f,sizeof(k)); for(i=1;i<=m;i++) x=s[i]=rd(),f[x][0]=g[x]=h[x][0]=k[x]=0,np=min(np,p[x]),mp=max(mp,p[x]); top=1; for(i=1;i<=n;i++) if(p[i]<=np&&q[i]>=mp&&dep[i]>=dep[top]) top=i; for(i=n;i>=2;i--) { x=Q[i],y=fa[x],d=dep[x]-dep[y]; if(f[y][0]>f[x][0]+d) f[y][1]=max(f[y][1],f[x][0]+d); else f[y][1]=f[y][0],f[y][0]=f[x][0]+d; if(!h[x][0]&&x!=top) h[y][0]=0,ans+=(dep[x]-dep[y])<<1; if(h[y][0]<h[x][0]+d) h[y][1]=min(h[y][1],h[x][0]+d); else h[y][1]=h[y][0],h[y][0]=h[x][0]+d; } for(i=2;i<=n;i++) { x=Q[i],y=fa[x],d=dep[x]-dep[y],g[x]=max(g[x],g[y]+d),k[x]=min(k[x],k[y]+d); if(f[y][0]==f[x][0]+d) g[x]=max(g[x],f[y][1]+d); else g[x]=max(g[x],f[y][0]+d); if(h[y][0]==h[x][0]+d) k[x]=min(k[x],h[y][1]+d); else k[x]=min(k[x],h[y][0]+d); } for(i=1;i<=n;i++) printf("%lld\n",ans+2*min(k[i],h[i][0])-max(g[i],f[i][0])); return 0; }
标签:出发点 led input printf ons for 必须 mil lld
原文地址:http://www.cnblogs.com/CQzhangyu/p/7514532.html