bzoj2599(点分治)

最新推荐文章于 2021-04-07 21:15:54 发布

qkoqhh

最新推荐文章于 2021-04-07 21:15:54 发布

阅读量241

点赞数

分类专栏：分治

本文链接：https://blog.csdn.net/qkoqhh/article/details/80747810

版权

分治专栏收录该内容

11 篇文章 0 订阅

订阅专栏

菜蛙一个错误debug半天qaq

其实很好写。。对没层分治而言，只需要统计到当前根的距离为i所需要的最少边数即可。。不过初始化稍微麻烦一点。。把子数中所有点都存下来然后直接把这几个点对应的距离初始化一下即可。。这样才能把复杂度降到O(nlogn)

然而还是跑得太慢了。。估计选根这里写的常熟有点大？

/**
 *　　　　　　  　　┏┓　　 　┏┓
 * 　　　　　  　　┏┛┗━━━━━━━┛┗━━━┓
 * 　　　　　  　　┃　　　　　　　┃ 　
 * 　　　　　  　　┃　　　━　　 　┃
 * 　　　　　  　　┃　＞　　　＜　┃
 * 　　　　　  　　┃　　　　　　　┃
 * 　　　　　  　　┃...　⌒　... 　┃
 * 　　　　　　  　┃              ┃
 * 　　　　　　  　┗━┓          ┏━┛
 * 　　　　　　　　　┃          ┃　Code is far away from bug with the animal protecting　　　　　　　　　　
 * 　　　　　　　　　┃          ┃   神兽保佑,代码无bug
 * 　　　　　　　　　┃          ┃　　　　　　　　　　　
 * 　　　　　　　　　┃          ┃  　　　　　　
 * 　　　　　　　　　┃          ┃
 * 　　　　　　　　　┃          ┃　　　　　　　　　　　
 * 　　　　　　　　　┃          ┗━━━┓
 * 　　　　　　　　　┃              ┣┓
 * 　　　　　　　　　┃              ┏┛
 * 　　　　　　　　　┗┓┓┏━━━━━━━━┳┓┏┛
 * 　　　　　　　　　　┃┫┫       ┃┫┫
 * 　　　　　　　　　　┗┻┛       ┗┻┛
 */ 
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<cmath>
#include<map>
#include<stack>
#include<set>
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,l,r) for(int i=l;i>=r;i--)
#define link(x) for(edge *j=h[x];j;j=j->next)
#define mem(a) memset(a,0,sizeof(a))
#define ll long long
#define eps 1e-12
#define succ(x) (1<<x)
#define lowbit(x) (x&(-x))
#define sqr(x) ((x)*(x))
#define mid (x+y>>1)
#define NM 200005
#define nm 400005
#define N 1000005
#define pi 3.1415926535897931
const int inf=1000000007;
using namespace std;
ll read(){
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}


struct edge{int t,v;edge*next;}e[nm],*h[NM],*o=e;
void add(int x,int y,int v){o->t=y;o->v=v;o->next=h[x];h[x]=o++;}
int n,m,_x,_y,_t,size[NM],tot,smin,root,b[NM],c[N],w[NM],ans,cnt;
ll d[NM];
bool v[NM];

void dfs1(int x,int f){size[x]=1;link(x)if(!v[j->t]&&j->t!=f)dfs1(j->t,x),size[x]+=size[j->t];}
void getroot(int x,int f){
    int s=tot-size[x];
    link(x)if(!v[j->t]&&j->t!=f)getroot(j->t,x),s=max(s,size[j->t]);
    if(s<smin)smin=s,root=x;
}

void dfs2(int x,int f){
    if(d[x]>m)return;
    b[++cnt]=x;ans=min(ans,w[x]+c[m-d[x]]);
    link(x)if(!v[j->t]&&j->t!=f){
	d[j->t]=d[x]+j->v;w[j->t]=w[x]+1;
	dfs2(j->t,x);
    }
}


void div(int x){
    dfs1(x,0);
    smin=inf;tot=size[x];
    getroot(x,0);
    v[root]++;cnt=0;
    //printf("%d\n",root);
    link(root)if(!v[j->t]){
	int t=cnt+1;
	d[j->t]=j->v;w[j->t]=1;
	dfs2(j->t,root);
	for(;t<=cnt;t++)if(d[b[t]]<=m)c[d[b[t]]]=min(c[d[b[t]]],w[b[t]]);
    }
    inc(i,1,cnt)if(d[b[i]]<=m)c[d[b[i]]]=inf;c[0]=0;
    link(root)if(!v[j->t])div(j->t);
}


int main(){
    n=read();m=read();ans=inf;
    inc(i,1,n-1){_x=1+read();_y=1+read();_t=read();add(_x,_y,_t);add(_y,_x,_t);}
    inc(i,1,m)c[i]=inf;
    div(1);
    if(ans<inf)printf("%d\n",ans);else printf("-1\n");
    return 0;
}