BZOJ_2599_[IOI2011]Race_点分治

BZOJ_2599_[IOI2011]Race_点分治

Description

给一棵树,每条边有权.求一条简单路径,权值和等于K,且边的数量最小.N <= 200000, K <= 1000000

Input

第一行 两个整数 n, k
第二..n行 每行三个整数 表示一条无向边的两端和权值 (注意点的编号从0开始)

Output

一个整数 表示最小边数量 如果不存在这样的路径 输出-1

Sample Input

Sample Output

2

点分治。开个桶记录一下长度为x的路径最小的深度。

如何去重？我们开一个辅助桶存一下根的当前儿子的子树内的贡献。

保证每次查的时候用的是其他儿子的子树的路径。

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 200050
#define inf 100000000
int head
,to[N<<1],nxt[N<<1],cnt,val[N<<1];
int root,sum,siz
,f
,g[1000050],d
,n,k,ans,dep
,tmp[1000050],a
,b
;
bool used
;
inline void add(int u,int v,int w) {
to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; val[cnt]=w;
}
void get_root(int x,int y) {
siz[x]=1; f[x]=0;
int i;
for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&!used[to[i]]) {
get_root(to[i],x);
siz[x]+=siz[to[i]];
f[x]=max(f[x],siz[to[i]]);
}
f[x]=max(f[x],sum-siz[x]);
if(f[x]<f[root]) root=x;
}
void get_dep(int x,int y) {
b[++b[0]]=x;
a[++a[0]]=x;
siz[x]=1;
dep[x]=dep[y]+1;
//printf("x=%d dep[x]=%d\n",x,dep[x]);
if(d[x]<=k) {
ans=min(ans,g[k-d[x]]+dep[x]-2);
tmp[d[x]]=min(tmp[d[x]],dep[x]);
}
int i;
for(i=head[x];i;i=nxt[i]) if(to[i]!=y&&!used[to[i]]) {
d[to[i]]=d[x]+val[i];
get_dep(to[i],x);
siz[x]+=siz[to[i]];
}
}
void work(int x) {
g[0]=1;
used[x]=1;
d[x]=0;
int i,j;
b[0]=0;
siz[x]=1;
dep[x]=1;
for(i=head[x];i;i=nxt[i]) if(!used[to[i]]) {
a[0]=0;
d[to[i]]=val[i];
get_dep(to[i],x);
siz[x]+=siz[to[i]];
for(j=1;j<=a[0];j++) if(d[a[j]]<=k) {
g[d[a[j]]]=min(g[d[a[j]]],tmp[d[a[j]]]);
}
for(j=1;j<=a[0];j++) if(d[a[j]]<=k) tmp[d[a[j]]]=inf;
}
for(i=1;i<=b[0];i++) if(d[b[i]]<=k) {
g[d[b[i]]]=tmp[d[b[i]]]=inf;
}
g[0]=tmp[0]=inf;
for(i=head[x];i;i=nxt[i]) if(!used[to[i]]) {
root=0;
sum=siz[to[i]];
get_root(to[i],0);
work(root);
}
}
int main() {
scanf("%d%d",&n,&k);
int i,x,y,z;
for(i=0;i<=k;i++) g[i]=tmp[i]=inf;
for(i=1;i<n;i++) {
scanf("%d%d%d",&x,&y,&z);
x++;y++;
add(x,y,z);add(y,x,z);
}
sum=n;
f[0]=inf;
ans=1<<30;
root=0;
get_root(1,0);
work(root);
printf("%d\n",ans>n?-1:ans);
}