BZOJ2599: [IOI2011]Race

Description

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

 

Input

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

Output

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

 

Code:

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82

// luogu-judger-enable-o2 #include<bits/stdc++.h> using namespace std; const int maxn=3000000; void setIO(string s) {     string in=s+".in",out=s+".out";     freopen(in.c_str(),"r",stdin);     // freopen(out.c_str(),"w",stdout); } int root,sn,edges,n,K,tl,tr,ans=1000000000; int f[maxn],siz[maxn],vis[maxn],hd[maxn],to[maxn],nex[maxn],val[maxn],dis1[maxn],dis2[maxn],mine[maxn];   void addedge(int u,int v,int c) {     nex[++edges]=hd[u],hd[u]=edges,to[edges]=v,val[edges]=c; } void getroot(int u,int ff) {     siz[u]=1,f[u]=0;     for(int i=hd[u];i;i=nex[i])     {         int v=to[i];         if(v==ff||vis[v]) continue;         getroot(v,u);         siz[u]+=siz[v];          f[u]=max(f[u],siz[v]);                }     f[u]=max(f[u],sn-siz[u]);     if(f[u]<f[root])root=u; } void getdis(int u,int ff,int d1,int d2) {     if(d1>K) return;     dis1[++tl]=d1,dis2[tl]=d2;     for(int i=hd[u];i;i=nex[i])     {         int v=to[i];         if(v==ff||vis[v]) continue;         getdis(v,u,d1+val[i],d2+1);      } } void dfs(int u) {     mine[0]=0,tl=0;     for(int i=hd[u];i;i=nex[i])     {         int v=to[i];         if(vis[v]) continue;         int pdl=tl;         getdis(v,u,val[i],1);                  for(int j=pdl+1;j<=tl;++j) ans=min(ans,mine[K-dis1[j]]+dis2[j]);         for(int j=pdl+1;j<=tl;++j) mine[dis1[j]]=min(mine[dis1[j]], dis2[j]);     }     for(int i=1;i<=tl;++i) mine[dis1[i]]=1000000000; } void Get(int u) {     vis[u]=1;     dfs(u);     for(int i=hd[u];i;i=nex[i])     {         int v=to[i];         if(vis[v]) continue;         root=0,sn=siz[v];         getroot(v,u);         Get(root);     } } int main() {     // setIO("input");     scanf("%d%d",&n,&K);     for(int i=1,u,v,c;i<n;++i)     {         scanf("%d%d%d",&u,&v,&c),++u,++v;         addedge(u,v,c),addedge(v,u,c);     }     for(int i=0;i<maxn;++i) mine[i]=1000000000;     vis[0]=1,sn=n,root=0,f[0]=100000000,getroot(1,0),Get(root);     printf("%d\n",ans>=n?-1:ans);     return 0; }