CH4908 Race

题意

4908 Race 0x49「数据结构进阶」练习

描述

给定一棵 N 个节点的树，每条边带有一个权值。

求一条简单路径，路径上各条边的权值和等于K，且路径包含的边的数量最少。

输入格式

第一行两个整数 N, K。
第2~N行每行三个整数x,y,z，表示一条无向边的两个端点x,y和权值z，点的编号从0开始。

输出格式

一个整数，表示最少边数量。如果不存在满足要求的路径，输出-1。

样例输入

4 3
0 1 1
1 2 2
1 3 4

样例输出

2

数据范围与约定

• N <= 200000, K <= 1000000

来源

IOI2011

        </article>

分析

训练指南的配套代码有问题，双指针绝对是错的。学习了hzwer的做法。

开一个100W的数组t,t[i]表示权值为i的路径最少边数

找到重心分成若干子树后， 得出一棵子树的所有点到根的权值和x，到根c条边，用t[k-x]+c更新答案，全部查询完后

然后再用所有c更新t[x]

这样可以保证不出现点分治中的不合法情况

把一棵树的所有子树搞完后再遍历所有子树恢复T数组，如果用memset应该会比较慢

时间复杂度\(O(n \log^2 n)\)

代码

网上好多代码，点分治递归的时候没有重新统计以重心为根的整棵树每个节点的size，按道理找出来的重心是错的，复杂度没有保证。

但是出题人根本就不会卡这种东西，所以不用写了，反而常数小。

#include<bits/stdc++.h>
#define rg register
#define il inline
#define co const
template<class T>il T read(){
    rg T data=0,w=1;rg char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') w=-1;ch=getchar();}
    while(isdigit(ch)) data=data*10+ch-'0',ch=getchar();
    return data*w;
}
template<class T>il T read(rg T&x) {return x=read<T>();}
typedef long long ll;
using namespace std;
typedef pair<int,int> pii;

co int N=2e5+1;
int n,k,ans,sum,root,max_size;
vector<pii> e[N];
int t[1000001],s[N],d[N],c[N],v[N];
void getroot(int x){
	v[x]=1,s[x]=1;
	int max_part=0;
	for(int i=0,y;i<e[x].size();++i){
		if(v[y=e[x][i].first]) continue;
		getroot(y);
		s[x]+=s[y],max_part=max(max_part,s[y]);
	}
	max_part=max(max_part,sum-s[x]);
	if(max_part<max_size) root=x,max_size=max_part;
	v[x]=0;
}
void cal(int x){
	v[x]=1;
	if(d[x]<=k) ans=min(ans,c[x]+t[k-d[x]]);
	for(int i=0,y;i<e[x].size();++i){
		if(v[y=e[x][i].first]) continue;
		c[y]=c[x]+1,d[y]=d[x]+e[x][i].second;
		cal(y);
	}
	v[x]=0;
}
void add(int x,int flag){
	v[x]=1;
	if(d[x]<=k){
		if(flag) t[d[x]]=min(t[d[x]],c[x]);
		else t[d[x]]=n;
	}
	for(int i=0,y;i<e[x].size();++i){
		if(v[y=e[x][i].first]) continue;
		add(y,flag);
	}
	v[x]=0;
}
void work(int x){
	v[x]=1,t[0]=0;
	for(int i=0,y;i<e[x].size();++i){
		if(v[y=e[x][i].first]) continue;
		c[y]=1,d[y]=e[x][i].second;
		cal(y),add(y,1);
	}
	for(int i=0,y;i<e[x].size();++i){
		if(v[y=e[x][i].first]) continue;
		add(y,0);
	}
	for(int i=0,y;i<e[x].size();++i){
		if(v[y=e[x][i].first]) continue;
		sum=max_size=s[y];
		getroot(y),work(root);
	}
}
int main(){
//	freopen(".in","r",stdin),freopen(".out","w",stdout);
	read(n),read(k);
	fill(t+1,t+k+1,n);
	for(int i=1,u,v,w;i<n;++i){
		read(u),read(v),read(w);
		e[++u].push_back(pii(++v,w)),e[v].push_back(pii(u,w));
	}
	ans=sum=max_size=n;
	getroot(1),work(root);
	if(ans==n) puts("-1");
	else printf("%d\n",ans);
	return 0;
}