BZOJ2599 IOI2011Race

题目：一棵树,每条边有权.求一条路径,权值和等于K,且边的数量最小.

点分治，我们考虑经过根节点的路径，t[x]表示路径长为x时最少的边数，然后每次拿栈记下来清空。

注意先搜索再更新，这样可以避免同一子树内互相到达。

By：大奕哥

 #include<bits/stdc++.h>
 using namespace std;
 const int N=2e5+,inf=1e9,M=1e6+;;
 int n,k,rt,cnt,head[N],s[N],f[N],t[M],ans,maxn,sta[N],top,dis[N],d[N];
 bool v[N];
 struct node{
     int to,nex,w;
 }e[N<<];
 void add(int x,int y,int w)
 {
     e[++cnt].w=w;e[cnt].nex=head[x];e[cnt].to=y;head[x]=cnt;
 }
 void getrt(int x,int fa)
 {
     s[x]=;f[x]=;
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(y==fa||v[y])continue;
         getrt(y,x);
         f[x]=max(f[x],s[y]);
         s[x]+=s[y];
     }
     f[x]=max(f[x],maxn-s[x]);
     if(f[x]<f[rt])rt=x;
 }
 void calc(int x,int fa)
 {
     if(dis[x]<=k)ans=min(ans,t[k-dis[x]]+d[x]);
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(v[y]||y==fa)continue;
         dis[y]=dis[x]+e[i].w;
         d[y]=d[x]+;
         calc(y,x);
     }
 }
 void update(int x,int fa)
 {
     if(dis[x]<=k)t[dis[x]]=min(t[dis[x]],d[x]),sta[++top]=dis[x];
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(v[y]||y==fa)continue;
         update(y,x);
     }
 }
 void work(int x)
 {
     v[x]=;top=;t[]=;
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(v[y])continue;
         dis[y]=e[i].w;d[y]=;
         calc(y,x);
         update(y,x);
     }
     for(int i=;i<=top;++i)t[sta[i]]=inf;
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(v[y])continue;
         maxn=s[y];rt=;
         getrt(y,x);
         work(rt);
     }
 }
 int main()
 {
     scanf("%d%d",&n,&k);
     int x,y,w;
     for(int i=;i<n;++i)
     {
         scanf("%d%d%d",&x,&y,&w);
         x++;y++;
         add(x,y,w);add(y,x,w);
     }
     memset(t,0x3f,sizeof(t));
     maxn=f[]=ans=n;
     getrt(,);
     work(rt);
     if(ans==n)puts("-1");
     else printf("%d\n",ans);
     return ;
 }