题目描述
给一棵树,每条边有权.求一条简单路径,权值和等于K,且边的数量最小.N <= 200000, K <= 1000000
输入
第一行 两个整数 n, k
第二..n行 每行三个整数 表示一条无向边的两端和权值 (注意点的编号从0开始)
输出
一个整数 表示最小边数量 如果不存在这样的路径 输出-1
样例输入
4 3
0 1 1
1 2 2
1 3 4
样例输出
2
题解
树的点分治
以前的做法太sb看着不爽自己把它卡掉了。。。
对树进行点分治，能够想到开桶维护距离当前根节点某距离的所有点中，最小的深度是多少。这样对于每一个点 $i$ ，直接用 $deep[i]+v[k-dis[i]]$ 更新答案即可。其中 $v$ 是桶。
但是最值并不满足可减性，因此不能容斥处理两个点在同一个子树内的情况。
考虑每次找儿子的过程中其实是有序的——选择当前子树内的节点和以前找到过的节点，这样是不重不漏的。
因此枚举所有子树，先统计子树贡献，然后再加到桶中。最后动态还原桶并分治子树部分。
注意：根节点的dis为0，但是树中存在0权边，可能会清掉v[0]，因此每次都要重新赋值v[0]=0。
使劲啊复杂度 $O(k+n\log n)$
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 200010
using namespace std;
int k , head[N] , to[N << 1] , len[N << 1] , next[N << 1] , cnt , si[N] , ms[N] , sum , root , deep[N] , dis[N] , vis[N] , v[N * 5] , ans = 1 << 30;
inline void add(int x , int y , int z)
{
to[++cnt] = y , len[cnt] = z , next[cnt] = head[x] , head[x] = cnt;
}
void getroot(int x , int fa)
{
int i;
si[x] = 1 , ms[x] = 0;
for(i = head[x] ; i ; i = next[i])
if(!vis[to[i]] && to[i] != fa)
getroot(to[i] , x) , si[x] += si[to[i]] , ms[x] = max(ms[x] , si[to[i]]);
ms[x] = max(ms[x] , sum - si[x]);
if(ms[x] < ms[root]) root = x;
}
void calc(int x , int fa)
{
int i;
if(dis[x] <= k) ans = min(ans , deep[x] + v[k - dis[x]]);
for(i = head[x] ; i ; i = next[i])
if(!vis[to[i]] && to[i] != fa)
deep[to[i]] = deep[x] + 1 , dis[to[i]] = dis[x] + len[i] , calc(to[i] , x);
}
void insert(int x , int fa)
{
int i;
if(dis[x] <= k) v[dis[x]] = min(v[dis[x]] , deep[x]);
for(i = head[x] ; i ; i = next[i])
if(!vis[to[i]] && to[i] != fa)
insert(to[i] , x);
}
void clear(int x , int fa)
{
int i;
if(dis[x] <= k) v[dis[x]] = 1 << 30;
for(i = head[x] ; i ; i = next[i])
if(!vis[to[i]] && to[i] != fa)
clear(to[i] , x);
}
void solve(int x)
{
int i;
vis[x] = 1 , v[0] = 0;
for(i = head[x] ; i ; i = next[i]) if(!vis[to[i]]) deep[to[i]] = 1 , dis[to[i]] = len[i] , calc(to[i] , 0) , insert(to[i] , 0);
for(i = head[x] ; i ; i = next[i]) if(!vis[to[i]]) clear(to[i] , 0);
for(i = head[x] ; i ; i = next[i]) if(!vis[to[i]]) sum = si[to[i]] , root = 0 , getroot(to[i] , 0) , solve(root);
}
int main()
{
int n , i , x , y , z;
scanf("%d%d" , &n , &k);
for(i = 1 ; i < n ; i ++ ) scanf("%d%d%d" , &x , &y , &z) , add(x + 1 , y + 1 , z) , add(y + 1 , x + 1 , z);
for(i = 1 ; i <= k ; i ++ ) v[i] = 1 << 30;
ms[0] = sum = n , getroot(1 , 0) , solve(root);
if(ans == 1 << 30) puts("-1");
else printf("%d\n" , ans);
return 0;
}
