Description
给一棵树,每条边有权.求一条简单路径,权值和等于K,且边的数量最小.N <= 200000, K <= 1000000
Input
第一行 两个整数 n, k 第二..n行 每行三个整数 表示一条无向边的两端和权值 (注意点的编号从0开始) Output
一个整数 表示最小边数量 如果不存在这样的路径 输出-1
点分治,维护 val[i] 表示到根的距离为 i <script type="math/tex" id="MathJax-Element-53">i</script>,最少需要几条边。对于每一棵子树,先一起更新答案,然后再一起维护。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int oo=0x3f3f3f3f;
int fir[200010],ne[400010],to[400010],w[400010],
size[200010],mx[200010],out[200010],val[1000010],last[1000010],
sw[200010],se[200010],
n,k,siz,r,ans=oo,clo,num;
void add(int num,int u,int v,int x)
{
ne[num]=fir[u];
fir[u]=num;
to[num]=v;
w[num]=x;
}
void init()
{
int u,v,x;
scanf("%d%d",&n,&k);
for (int i=1;i<n;i++)
{
scanf("%d%d%d",&u,&v,&x);
add(i*2,u,v,x);
add(i*2+1,v,u,x);
}
}
void dfs1(int u,int fa)
{
int v;
mx[u]=-1;
size[u]=1;
for (int i=fir[u];i;i=ne[i])
if (!out[v=to[i]]&&v!=fa)
{
dfs1(v,u);
size[u]+=size[v];
mx[u]=max(mx[u],size[v]);
}
mx[u]=max(mx[u],siz-size[u]);
if (r==-1||mx[u]<mx[r]) r=u;
}
void dfs2(int u,int ww,int ee,int fa)
{
size[u]=1;
int v;
if (ww<=k)
{
sw[++num]=ww;
se[num]=ee;
}
for (int i=fir[u];i;i=ne[i])
if (!out[v=to[i]]&&v!=fa)
{
dfs2(v,ww+w[i],ee+1,u);
size[u]+=size[v];
}
}
void solve(int u)
{
int v;
r=-1;
dfs1(u,-1);
last[val[0]=0]=++clo;
for (int i=fir[r];i;i=ne[i])
if (!out[v=to[i]])
{
num=0;
dfs2(v,w[i],1,r);
for (int j=1;j<=num;j++)
if (last[k-sw[j]]==clo)
ans=min(ans,val[k-sw[j]]+se[j]);
for (int j=1;j<=num;j++)
if (last[sw[j]]<clo||val[sw[j]]>se[j])
{
last[sw[j]]=clo;
val[sw[j]]=se[j];
}
}
out[r]=1;
for (int i=fir[r];i;i=ne[i])
if (!out[v=to[i]])
{
siz=size[v];
solve(v);
}
}
int main()
{
init();
if (k==0)
{
printf("0\n");
return 0;
}
siz=n;
solve(0);
printf("%d\n",ans==oo?-1:ans);
}