BZOJ_P2599 [IOI2011]Race(点分治)

BZOJ传送门
Time Limit: 50 Sec Memory Limit: 128 MB
Submit: 2190 Solved: 653
[Submit][Status][Discuss]
Description
给一棵树,每条边有权.求一条路径,权值和等于K,且边的数量最小.

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

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

Sample Input
4 3
0 1 1
1 2 2
1 3 4

Sample Output
2

HINT
Source

卡评测好题！卡了三次评测QuQ

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<iostream>
using namespace std;
#define N 200005
inline int in(int x=0,int v=1,char ch=getchar()){
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();if(ch=='-') v=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x*v;
}
struct Node{int v,l;};vector<Node> g[N];
int n,k,ans,sz,root,s[N],d[N],f[N],cnt,t[N*5];
bool b[N];int deep[N];
void GetRoot(int u,int fa){
    s[u]=1,f[u]=0;
    for(int i=0,v,lim=g[u].size();i<lim;i++)
        if((v=g[u][i].v)!=fa&&!b[v])
            GetRoot(v,u),s[u]+=s[v],f[u]=max(f[u],s[v]);
    f[u]=max(f[u],sz-s[u]);if(f[u]<f[root]) root=u;
}
void GetDeep(int u,int fa){
    if(deep[u]<=k) ans=min(ans,d[u]+t[k-deep[u]]);
    for(int i=0,v,lim=g[u].size();i<lim;i++)
        if((v=g[u][i].v)!=fa&&!b[v])
            d[v]=d[u]+1,deep[v]=deep[u]+g[u][i].l,GetDeep(v,u);
}
void Calc(int u,int fa,bool typ){
    if(deep[u]<=k)
        if(typ) t[deep[u]]=min(t[deep[u]],d[u]);
        else t[deep[u]]=N;
    for(int i=0,v,lim=g[u].size();i<lim;i++)
        if((v=g[u][i].v)!=fa&&!b[v]) Calc(v,u,typ);
}
void GetAns(int u){
    b[u]=1,t[0]=0;
    for(int i=0,v,lim=g[u].size();i<lim;i++)
        if(!b[v=g[u][i].v])
            d[v]=1,deep[v]=g[u][i].l,GetDeep(v,0),Calc(v,0,1);
    for(int i=0,v,lim=g[u].size();i<lim;i++) if(!b[v=g[u][i].v]) Calc(v,0,0);
    for(int i=0,v,lim=g[u].size();i<lim;i++)
        if(!b[v=g[u][i].v])
            sz=s[v],GetRoot(v,root=0),GetAns(root);
}
inline void init(){memset(b,0,sizeof(b));for(int i=0;i<=n;i++) g[i].clear();
    memset(t,0x7f,sizeof(t));f[0]=sz=n,root=ans=0;}
int main(){
    n=in(),k=in();init();ans=N;
    for(int i=1,x,y,z;i<n;i++)
        x=in()+1,y=in()+1,z=in(),
        g[x].push_back((Node){y,z}),
        g[y].push_back((Node){x,z});
    GetRoot(1,root);GetAns(root);
    printf("%d\n",ans==N?-1:ans);
    return 0;
}