码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj2599

时间:2016-05-07 20:54:29      阅读:185      评论:0      收藏:0      [点我收藏+]

标签:

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*2],v[N*2],val[N*2],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=0; char ch; bool bo=1;
    while (ch=getchar(),ch<0||ch>9) if (ch==-) bo=0;
    while (x=x*10+ch-0,ch=getchar(),ch>=0&&ch<=9);
    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]=1; f[u]=0;
    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+1;
        getarray(son,u);
    }
    
}
void calc(int u,int value, int f)
{
    d[u].val=value; if (f==1) d[u].size=0; else d[u].size=1; 
    sum=0; da[++sum].val=value; da[sum].size=d[u].size;
    getarray(u,0);
    sort(da+1,da+1+sum,cmp);
    for (int i=1,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]=1; calc(u,0,1);
    for    (int p=now[u]; p; p=pre[p])
    {
        int son=v[p];
        if (vis[son]) continue;
        calc(son,val[p],-1);
        all=size[son]; 
        root=0; getroot(son,0);
        solve(root);
    }
}
int main()
{
    n=read(); k=read();
    for (int i=1; i<n; i++)
    {
        int u=read()+1,v=read()+1,value=read();
        ins(u,v,value); ins(v,u,value);
    }
    all=n; f[root=0]=n; getroot(1,0); 
    solve(root);
    for (int i=1; i<n; i++)
    {
        if (ans[i]) 
        {
            printf("%d\n",i);
            return 0;
        }
    }
    printf("-1\n"); return 0;
} 
View Code

 

bzoj2599

标签:

原文地址:http://www.cnblogs.com/HQHQ/p/5469111.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!