BZOJ2599: [IOI2011]Race

最新推荐文章于 2020-01-13 20:23:43 发布

CR1SceNT

最新推荐文章于 2020-01-13 20:23:43 发布

阅读量211

点赞数

分类专栏：点分治

本文链接：https://blog.csdn.net/Ep1C_HeReT1c/article/details/53981527

版权

点分治专栏收录该内容

5 篇文章 0 订阅

订阅专栏

题目链接

%%%http://hzwer.com/4286.html

【代码】

#include <iostream>  
#include <cstdio> 
#include <string> 
#include <cstring>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <algorithm>
#include <set>
#include <cmath>
#define N 200010
#define INF 1000000000
#define eps 1e-6
using namespace std;
typedef long long LL;
int n,k;
struct edge{
    int b,nt,w;
}e[N*2];
int p[N],nn;
void anode(int x,int y,int z){
    nn++;e[nn].b=y;e[nn].nt=p[x];p[x]=nn;e[nn].w=z;
    nn++;e[nn].b=x;e[nn].nt=p[y];p[y]=nn;e[nn].w=z;
}

int cnt[N*5];
int sz[N],f[N],sum,root,flag[N];
void getroot(int x,int fa){
    sz[x]=1;f[x]=0;
    for(int i=p[x];i;i=e[i].nt){
        int t=e[i].b;
        if(t==fa||flag[t])continue;
        getroot(t,x);sz[x]+=sz[t];f[x]=max(f[x],sz[t]);
    }f[x]=max(f[x],sum-sz[x]);
    if(f[x]<f[root])root=x;
}
int deep[N],dis[N];
void getd(int x,int dp,int d,int fa){
    deep[++deep[0]]=dp;dis[++dis[0]]=d;
    for(int i=p[x];i;i=e[i].nt){
        int t=e[i].b;
        if(t==fa||flag[t])continue;
        getd(t,dp+1,d+e[i].w,x);
    }
}
int ans=1000000000;

void Erase(int x,int d,int fa){
    if(d>k)return;
    cnt[d]=1000000000;
    for(int i=p[x];i;i=e[i].nt){
        int t=e[i].b;
        if(flag[t]||t==fa)continue;
        Erase(t,d+e[i].w,x);
    }
}

void solve(int x){
    flag[x]=1;
    for(int i=p[x];i;i=e[i].nt){
        int t=e[i].b;deep[0]=dis[0]=0;cnt[0]=0;
        if(flag[t])continue;
        getd(t,1,e[i].w,x);
        for(int j=1;j<=deep[0];j++)
            if(dis[j]<=k)ans=min(ans,cnt[k-dis[j]]+deep[j]);
        for(int j=1;j<=deep[0];j++)
            if(dis[j]<=k)cnt[dis[j]]=min(cnt[dis[j]],deep[j]);
    }
    for(int i=p[x];i;i=e[i].nt){
        int t=e[i].b;
        if(flag[t])continue;
        Erase(t,e[i].w,x);
    }
    for(int i=p[x];i;i=e[i].nt){
        int t=e[i].b;
        if(flag[t])continue;
        root=0;sum=sz[t];
        getroot(t,x);
        solve(root);
    } 
}

int main()  
{  
    scanf("%d%d",&n,&k);
    sum=n;f[0]=1e9;
    for(int i=1;i<=k;i++)cnt[i]=1e9;
    for(int i=1;i<n;i++){
        int aa,bb,cc;scanf("%d%d%d",&aa,&bb,&cc);
        aa++;bb++;anode(aa,bb,cc);
    } 
    getroot(1,0);
    solve(root);
    if(ans==1e9)ans=-1;
    printf("%d\n",ans);
    return 0;  
}