luoguP1084 疫情控制(题解)(搜索+贪心)
#include<iostream> #include<cstdlib> #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> #define il inline #define rg register #define ll long long #define N 60000 #define inf 2147483647 using namespace std; int n,m,hd[N],cnt,u,v,w; int c,idx[N]; int le,ri=500000; int dis[N][20],up[N][20]; int ok[N],use[N],vis[N]; struct S{ int res,idx; }ne[N],to[N]; struct T{ int nt,to,w; }edge[N<<1]; il void re(rg int &x); void add(rg int fm,rg int to,rg int w); void ST(rg int i,rg int fa); int check(rg int lim); int Dfs(rg int u,rg int fa); int Cmp(const S &x,const S &y); int main() { re(n); for(rg int i=1;i<n;++i) { re(u),re(v),re(w); add(u,v,w),add(v,u,w); if(u==1||v==1)c++; } re(m); if(m<c){cout<<-1;return 0;} for(rg int i=1;i<=m;++i) re(idx[i]); ST(1,0); for(rg int k=1;k<=19;++k) { for(rg int i=1;i<=n;++i) { up[i][k]=up[up[i][k-1]][k-1]; dis[i][k]=dis[up[i][k-1]][k-1]+dis[i][k-1]; } } while(le<ri) { rg int mid=((le+ri)>>1); if(check(mid))ri=mid; else le=mid+1; } cout<<le; return 0; } il void re(rg int &x) { rg int res=0;rg int w=1;char c=getchar(); while((c<'0'||c>'9')&&c!='-')c=getchar(); if(c=='-')w=-1,c=getchar(); while(c>='0'&&c<='9')res=(res<<3)+(res<<1)+c-'0',c=getchar(); x=w*res; } void add(rg int fm,rg int to,rg int w) { edge[++cnt].nt=hd[fm]; edge[cnt].to=to; edge[cnt].w=w; hd[fm]=cnt; } void ST(rg int i,rg int fa) { up[i][0]=fa; for(rg int k=hd[i];k;k=edge[k].nt) { rg int qw=edge[k].to; if(qw==fa)continue; dis[qw][0]=edge[k].w; ST(qw,i); } } int check(rg int lim){ memset(ok,0,sizeof(ok)); memset(use,0,sizeof(use)); memset(vis,0,sizeof(vis)); rg int cnt1=0,cnt2=0; rg int top=1; for(rg int i=1;i<=m;++i){ rg int res=lim,u=idx[i]; for(rg int k=19;k>=0;--k){ if(up[u][k]&&up[u][k]!=1&&res>=dis[u][k]) res-=dis[u][k],u=up[u][k]; } if(up[u][0]==1&&res>dis[u][0]) { to[++cnt1].res=res-dis[u][0]; to[cnt1].idx=u; } else ok[u]=1; } Dfs(1,0); for(rg int k=hd[1];k;k=edge[k].nt){ rg int qw=edge[k].to; if(!ok[qw]){ ne[++cnt2].res=dis[qw][0]; ne[cnt2].idx=qw; } } sort(to+1,to+cnt1+1,Cmp); sort(ne+1,ne+cnt2+1,Cmp); if(cnt1<cnt2)return 0; for(rg int i=1;i<=cnt1;++i) { if(!vis[to[i].idx]) { vis[to[i].idx]=1; continue; } while(vis[ne[top].idx])top++; if(to[i].res>=ne[top].res) { vis[ne[top].idx]=1,top++; continue; } if(top>cnt2)break; } while(vis[ne[top].idx])top++; if(top<=cnt2)return 0; return 1; } int Dfs(rg int u,rg int fa) { rg int temp=1,flag=0; for(rg int k=hd[u];k;k=edge[k].nt) { rg int qw=edge[k].to; if(qw==fa)continue; flag++,temp&=Dfs(qw,u); } if(!flag)temp=0; ok[u]=(ok[u]|temp); if(ok[u]&&up[u][0]==1) vis[u]=1; return ok[u]; } int Cmp(const S &x,const S &y){return x.res<y.res;}
哪怕人间是炼狱,梦想永远是天堂
继续走下去吧,理想永远都年轻,花儿一定会再次盛开