标签：idx fa int void cf 差分 dep 树上

[模板题]https://www.luogu.com.cn/problem/P3128

在(u,v)最短路径上每个点加上v,询问最后权值最大的点

const int N = 5e4 + 50,M = 1e5 +50;
int h[N],e[M],ne[M],idx;
void add(int a,int b) {
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

int n,m,root;
int dep[N],f[N][31];//i结点的2^j辈祖先
int cf[N];//差分数组,记录结点的子树权值和

void dfs(int u,int fa) {//预处理
	dep[u]=dep[fa]+1;
	for(int i=0; i<30; i++) {  //注意这里不取等号
		f[u][i+1]=f[f[u][i]][i];
	}
	for(int i=h[u]; ~i; i=ne[i]) {
		int j=e[i];
		if(j == fa)continue;
		f[j][0]=u;
		dfs(j,u);
	}
}

int lca(int x,int y) {
	if(dep[x]<dep[y])swap(x,y);
	for(int i=30; i>=0; i--) {   //x,y跳到同一深度
		if(dep[f[x][i]]>=dep[y])x=f[x][i];
		if(x == y)return x;
	}
	for(int i=30; i>=0; i--) {
		if(f[x][i]!=f[y][i]) {
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];//共同的fa
}

int maxn;
void find_ans(int u,int fa) {
	for(int i=h[u]; ~i; i=ne[i]) {
		int j = e[i];
		if(j == fa)continue;
		find_ans(j,u);
		cf[u] += cf[j];   //cf[u]存u结点所有子树的权值和
	}
	maxn = max(maxn,cf[u]);
}

void work() {
	scanf("%d%d",&n,&m);
	memset(h,-1,sizeof(h));
	for(int i=1; i<n; i++) {
		int u,v;scanf("%d%d",&u,&v);
		add(u,v);add(v,u);
	}
	dfs(1,0);
	while(m--) {
		int u,v;
		scanf("%d%d",&u,&v);
		int ans = lca(u,v);
		cf[u]++;   //点权差分
		cf[v]++;
		cf[ans]--;
		cf[f[ans][0]]--;
		/*边权差分
		cf[u]+=1;
		cf[v]+=1;
		cf[ans]-=2;
		*/
	}
	maxn = 0;
	find_ans(1,-1);
	printf("%d\n",maxn);
}

标签：idx,fa,int,void,cf,差分,dep,树上
来源： https://www.cnblogs.com/LaiYiC/p/15236503.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。