题目描述
给一棵树,每条边有权.求一条简单路径,权值和等于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; }