点分治裸题
#include<iostream> #include<cstring> #include<cstdio> using namespace std; inline int rd(){ int ret=0,f=1;char c; while(c=getchar(),!isdigit(c))f=c=='-'?-1:1; while(isdigit(c))ret=ret*10+c-'0',c=getchar(); return ret*f; } const int MAXN=200005; struct Edge{ int next,to,w; }e[MAXN<<1]; int ecnt,head[MAXN]; inline void add(int x,int y,int w){ e[++ecnt].next = head[x]; e[ecnt].to = y; e[ecnt].w = w; head[x] = ecnt; } int n,m; bool vis[MAXN]; int siz[MAXN]; void getsiz(int x,int pre){ siz[x]=1; for(int i=head[x];i;i=e[i].next){ int v=e[i].to; if(v==pre||vis[v]) continue; getsiz(v,x); siz[x]+=siz[v]; } } int mn,root; void getroot(int x,int pre,int tot){ int mx=0; for(int i=head[x];i;i=e[i].next){ int v=e[i].to; if(v==pre||vis[v]) continue; mx=max(mx,siz[v]); getroot(v,x,tot); } mx=max(mx,tot-siz[x]); if(mx<mn) mn=mx,root=x; } int s[MAXN],l[MAXN],sav[MAXN]; int f[1000005],g[1000005]; void dfs(int x,int pre,int dis,int len){ s[++s[0]]=dis;l[++l[0]]=len;sav[++sav[0]]=dis; for(int i=head[x];i;i=e[i].next){ int v=e[i].to; if(vis[v]||v==pre) continue; dfs(v,x,dis+e[i].w,len+1); } } int ans=0x3f3f3f3f; void dac(int x){ g[0]=0;f[0]=1;sav[0]=0;mn=n; getsiz(x,-1); getroot(x,-1,siz[x]); int u=root;vis[u]=1; for(int i=head[u];i;i=e[i].next){ int v=e[i].to; if(vis[v]) continue; l[0]=0;s[0]=0;dfs(v,x,e[i].w,1); for(int j=s[0];j>=1;j--){ if(s[j]>m) continue; if(f[m-s[j]]) ans=min(ans,g[m-s[j]]+l[j]); } for(int j=s[0];j>=1;j--){ if(s[j]>m) continue;//%%%%Monster_Qi f[s[j]]=1; g[s[j]]=min(g[s[j]],l[j]); } } for(int i=sav[0];i>=1;i--) if(sav[i]<=m) f[sav[i]]=0,g[sav[i]]=0x3f3f3f3f; for(int i=head[u];i;i=e[i].next){ int v=e[i].to; if(!vis[v]) dac(v); } } int main(){ memset(g,0x3f,sizeof(g)); n=rd();m=rd(); int x,y,w; for(int i=1;i<n;i++){ x=rd();y=rd();w=rd(); add(x,y,w);add(y,x,w); } dac(1); if(ans>=0x3f3f3f3f) cout<<-1; else cout<<ans; return 0; }