[BZOJ2599]Race

Description

给一棵树,每条边有权.求一条简单路径,权值和等于K,且边的数量最小.N <= 200000, K <= 1000000

Input

第一行 两个整数 n, k
第二..n行 每行三个整数 表示一条无向边的两端和权值 (注意点的编号从0开始)

Output

一个整数 表示最小边数量 如果不存在这样的路径 输出-1

Sample Input

4 3
0 1 1
1 2 2
1 3 4

Sample Output

2

HINT

2018.1.3新加数据一组，未重测

Source

很简单的一道点分治，注意一下维护的顺序，先更新答案再更新桶数组

代码：

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #define M 200010
 using namespace std;
 int n,m,num,rt,ans,k,S;
 int maxn[M],head[M],dis[M],d[M],size[M],t[M<<];
 bool vis[M];
 struct point{int to,next,dis;}e[M<<];
 void add(int from,int to,int dis) {
     e[++num].next=head[from];
     e[num].to=to;
     e[num].dis=dis;
     head[from]=num;
 }
 void getroot(int x,int fa) {
     size[x]=;maxn[x]=;
     for(int i=head[x];i;i=e[i].next) {
         int to=e[i].to;
         if(to==fa||vis[to]) continue;
         getroot(to,x),size[x]+=size[to];
         maxn[x]=max(maxn[x],size[to]);
     }
     maxn[x]=max(maxn[x],S-size[x]);
     if(maxn[x]<maxn[rt]) rt=x;
 }
 void cal(int x,int fa) {
     if(dis[x]>k) return;
     ans=min(ans,d[x]+t[k-dis[x]]);
     for(int i=head[x];i;i=e[i].next) {
         int to=e[i].to;
         if(vis[to]||to==fa) continue;
         d[to]=d[x]+,dis[to]=dis[x]+e[i].dis;
         cal(to,x);
     }
 }
 void insert(int x,int fa) {
     if(dis[x]<=k) {
         t[dis[x]]=min(t[dis[x]],d[x]);
         for(int i=head[x];i;i=e[i].next)
             if(!vis[e[i].to]&&e[i].to!=fa)
                 insert(e[i].to,x);
     }
 }
 void del(int x,int fa) {
     if(dis[x]<=k) {
         t[dis[x]]=1e9;
         for(int i=head[x];i;i=e[i].next)
             if(!vis[e[i].to]&&e[i].to!=fa)
                 del(e[i].to,x);
     }
 }
 void solve(int x) {
     vis[x]=true;t[]=;
     for(int i=head[x];i;i=e[i].next) {
         int to=e[i].to;
         if(vis[to]) continue;
         d[to]=,dis[to]=e[i].dis,cal(to,);
         insert(to,x);
     }
     for(int i=head[x];i;i=e[i].next)
         if(!vis[e[i].to])
             del(e[i].to,x);
     for(int i=head[x];i;i=e[i].next) {
         int to=e[i].to;
         if(vis[to]) continue;
         rt=,S=size[to],getroot(to,);
         solve(rt);
     }
 }
 int main() {
     scanf("%d%d",&n,&k);ans=n+;
     memset(t,,sizeof(t));
     for(int i=;i<n;i++) {
         int a,b,c;scanf("%d%d%d",&a,&b,&c);
         a++,b++;
         add(a,b,c),add(b,a,c);
     }
     S=maxn[]=n;getroot(,);
     solve(rt);
     if(ans>n) puts("-1");
     else printf("%d\n",ans);
     return ;
 }