【点分治】BZOJ 2599:[IOI2011]Race

BZOJ 2599:[IOI2011]Race

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

N<=200000 K<=1000000

Solution

t[x]表示以当前root为根,离根距离为x的点经过的最小边数.

然后每次找一颗子树cal,然后再把它add进去

最后清空那里听hzwer说好像这样比memset要快？

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

using namespace std;

#define maxn 2000001

struct edge{
int to,lst,c;
}e[maxn*2];

int last[maxn],tot,vis[maxn],son[maxn],size,root,d[maxn],t[maxn],m,ans,f[maxn];

long long dis[maxn];

inline int in()
{
int x=0;char ch=getchar();
while(ch<'0'||ch>'9')ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-'0',ch=getchar();
return x;
}

void Add(int u,int v,int c)
{
e[++tot]=(edge){v,last[u],c};last[u]=tot;
e[++tot]=(edge){u,last[v],c};last[v]=tot;
}

void findroot(int poi,int lst)
{
son[poi]=1;f[poi]=0;
for(int i=last[poi];i;i=e[i].lst)
if(e[i].to!=lst && !vis[e[i].to])
{
findroot(e[i].to,poi);
son[poi]+=son[e[i].to];
f[poi]=max(f[poi],son[e[i].to]);
}
f[poi]=max(f[poi],size-son[poi]);
if(f[poi]<f[root])root=poi;
}

void cal(int poi,int lst)
{
if(dis[poi]<=m)ans=min(ans,t[m-dis[poi]]+d[poi]);
for(int i=last[poi];i;i=e[i].lst)
if(e[i].to!=lst && !vis[e[i].to])
{
d[e[i].to]=d[poi]+1;
dis[e[i].to]=dis[poi]+e[i].c;
cal(e[i].to,poi);
}
}

void add(int poi,int lst,int flag)
{
if(dis[poi]<=m)
{
if(!flag)t[dis[poi]]=min(t[dis[poi]],d[poi]);
else t[dis[poi]]=99999999;
}
for(int i=last[poi];i;i=e[i].lst)
if(e[i].to!=lst && !vis[e[i].to])
add(e[i].to,poi,flag);
}

void work(int poi)
{
vis[poi]=1;t[0]=0;
for(int i=last[poi];i;i=e[i].lst)
if(!vis[e[i].to])
{
d[e[i].to]=1,dis[e[i].to]=e[i].c;
cal(e[i].to,0);
add(e[i].to,0,0);
}
for(int i=last[poi];i;i=e[i].lst)
if(!vis[e[i].to])
add(e[i].to,0,1);
for(int i=last[poi];i;i=e[i].lst)
if(!vis[e[i].to])
{
root=0;size=son[e[i].to];
findroot(e[i].to,poi);
work(root);
}
}

int main()
{
freopen("2599.in","r",stdin);
int n;
n=in(),m=in();ans=n+1;
for(int i=1;i<=m;i++)t[i]=n;
for(int i=1;i<n;i++)
{
int u=in()+1,v=in()+1,c=in();
Add(u,v,c);
}
size=n;f[0]=n;
root=0;
findroot(1,0);
work(root);
if(ans==n+1)printf("-1");
else printf("%d",ans);
return 0;
}