【bzoj2599】[IOI2011]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

题解
点分治，两边dfs找重心超时了，仿照黄学长的打法居然快了这么多。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#define inf 1000000000
int n,k,tot,sum,rt,ans;
int Next[400005],len[400005],ret[400005];
int t[1000005],Head[200005],f[200005],size[200005],dis[200005],d[200005];
bool flag[200005];
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void ins(int u,int v,int l)
{
    tot++;ret[tot]=v;len[tot]=l;Next[tot]=Head[u];Head[u]=tot;
}
void pre(int u,int f)
{
    size[u]=1;
    for (int i=Head[u];i;i=Next[i])
    {
        int v=ret[i];
        if (v!=f&&!flag[v])
        {
            pre(v,u);
            size[u]+=size[v];
        }
    }
}
void getroot(int x,int fa)
{
    size[x]=1;f[x]=0;
    for (int i=Head[x];i;i=Next[i])
    {
        if (ret[i]!=fa&&!flag[ret[i]])
        {
            getroot(ret[i],x);
            size[x]+=size[ret[i]];
            f[x]=max(f[x],size[ret[i]]);
        }
    }
    f[x]=max(f[x],sum-size[x]);
    if (f[x]<f[rt]) rt=x;
}
void cal(int x,int f)
{
    if (dis[x]<=k)ans=min(ans,d[x]+t[k-dis[x]]);
    for (int i=Head[x];i;i=Next[i])
    {
        int v=ret[i];
        if (v!=f&&!flag[v])
        {
            d[v]=d[x]+1;
            dis[v]=dis[x]+len[i];
            cal(v,x);
        }
    }
}
void add(int x,int f,bool fl)
{
    if (dis[x]<=k)
    {
        if (fl) t[dis[x]]=min(t[dis[x]],d[x]);
        else t[dis[x]]=inf;
    }
    for (int i=Head[x];i;i=Next[i])
    {
        int v=ret[i];
        if (v!=f&&!flag[v])
            add(v,x,fl);
    }
}
void solve(int RT)
{
    flag[RT]=1;t[0]=0;
    for (int i=Head[RT];i;i=Next[i])
    {
        int v=ret[i];
        if (!flag[v])
        {
            d[v]=1;dis[v]=len[i];
            cal(v,0);
            add(v,0,1);
        }
    }
    for (int i=Head[RT];i;i=Next[i])
    {
        if (!flag[ret[i]])
            add(ret[i],0,0);
    }
    for (int i=Head[RT];i;i=Next[i])
    {
        if (!flag[ret[i]])
        {
            rt=0;sum=size[ret[i]];
            getroot(ret[i],0);
            solve(rt);
        }
    }
}
int main()
{
    memset(t,127,sizeof(t));
    n=read();k=read();
    for (int i=1;i<n;i++)
    {
        int u=read()+1,v=read()+1,l=read();
        ins(u,v,l);ins(v,u,l);
    }
    sum=f[0]=ans=n;
    getroot(1,0);
    solve(rt);
    if (ans!=n) cout<<ans;
    else puts("-1");
}