bzoj2599

2599: [IOI2011]Race

Time Limit: 70 Sec  Memory Limit: 128 MB
Submit: 2476  Solved: 733
[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

 题解：

　　随便点分治，用一个ans[i]代表u,v距离为k，边数为i的点对个数，那么答案就为i （ans[i]>0 && i 最小）；

　　怎么求ans？ 自己yy一下吧

代码：

　　

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#define N 200100
using namespace std;
int tot,n,m,all,k,root,sum;
int pre[N*],v[N*],val[N*],now[N],size[N],ans[N],f[N];
bool vis[N];
struct data{int val,size;
}d[N],da[N];
bool cmp(data a,data b)
{
    return a.val<b.val;
}
int read()
{
    int x=; char ch; bool bo=;
    while (ch=getchar(),ch<''||ch>'') if (ch=='-') bo=;
    while (x=x*+ch-'',ch=getchar(),ch>=''&&ch<='');
    if (!bo) return -x; return x;
}
void ins(int a,int b,int c)
{
    ++tot; pre[tot]=now[a]; now[a]=tot; v[tot]=b; val[tot]=c;
}
void getroot(int u,int fa)
{
    size[u]=; f[u]=;
    for (int p=now[u]; p; p=pre[p])
    {
        int son=v[p];
        if (vis[son]||son==fa) continue;
        getroot(son,u);
        size[u]+=size[son];
        f[u]=max(f[u],size[son]);
    }
    f[u]=max(f[u],all-size[u]);
    if (f[u]<f[root]) root=u;
}
void getarray(int u,int fa)
{
    for (int p=now[u]; p; p=pre[p])
    {
        int son=v[p];
        if (son==fa||vis[son]) continue;
        da[++sum].val=d[son].val=d[u].val+val[p];
        da[sum].size=d[son].size=d[u].size+;
        getarray(son,u);
    }

}
void calc(int u,int value, int f)
{
    d[u].val=value; if (f==) d[u].size=; else d[u].size=;
    sum=; da[++sum].val=value; da[sum].size=d[u].size;
    getarray(u,);
    sort(da+,da++sum,cmp);
    for (int i=,j=sum; i<=j; i++)
    {
        while (j>i && da[j].val+da[i].val>k) j--;
        for (int p=j; da[i].val+da[p].val==k ;p--)ans[da[p].size+da[i].size]+=f;
    }
}
void solve(int u)
{
    vis[u]=; calc(u,,);
    for    (int p=now[u]; p; p=pre[p])
    {
        int son=v[p];
        if (vis[son]) continue;
        calc(son,val[p],-);
        all=size[son];
        root=; getroot(son,);
        solve(root);
    }
}
int main()
{
    n=read(); k=read();
    for (int i=; i<n; i++)
    {
        int u=read()+,v=read()+,value=read();
        ins(u,v,value); ins(v,u,value);
    }
    all=n; f[root=]=n; getroot(,);
    solve(root);
    for (int i=; i<n; i++)
    {
        if (ans[i])
        {
            printf("%d\n",i);
            return ;
        }
    }
    printf("-1\n"); return ;
}