CF1000G Two-Paths&BSOJ6376【11.05模拟】草莓

前言

这个题还有个加强版:BSOJ6376【11.05模拟】草莓

CF1000G Two-Paths&BSOJ6376【11.05模拟】草莓

大概是人均场切了。

题目

CF1000G Two-Paths

CF1000G Two-Paths&BSOJ6376【11.05模拟】草莓

分析

首先这个路径一定是一条链加上若干回路的路径。

每次 \(x\) 到 \(y\) 的途中都可以进入子树进行获取贡献,为了尽可能大,我们可以先 \(dp\) 出每一个点到子树再回来可以得到的最大贡献。

以及每一个点到父亲再回来可以得到的最大贡献。

再处理每一个点对父亲的 \(dp\) 值的贡献,那么每次询问就变成了询问路径和,使用树剖维护即可。

对于加强版就是还要考虑来回的路径值不同,这个可以前缀和一下解决。

代码

#include<bits/stdc++.h>
using namespace std;
template <typename T>
inline void read(T &x){
	x=0;char ch=getchar();bool f=false;
	while(!isdigit(ch)){if(ch=='-'){f=true;}ch=getchar();}
	while(isdigit(ch)){x=x*10+(ch^48);ch=getchar();}
	x=f?-x:x;
	return ;
}
template <typename T>
inline void write(T x){
	if(x<0) putchar('-'),x=-x;
	if(x>9) write(x/10);
	putchar(x%10^48);
	return ;
}
const int N=5e5+5;
#define ll long long
int n,a[N],q,idx;
int head[N],to[N<<1],val1[N<<1],val2[N<<1],nex[N<<1],fa[N][20],dep[N];
inline void add(int u,int v,int w1,int w2){
	nex[++idx]=head[u];
	to[idx]=v;
	head[u]=idx;
	val1[idx]=w1;
	val2[idx]=w2;
	return ;
}
ll dp[N],dp1[N],dpp[N],c[N],costd[N],costu[N],c1[N],num[N];
void dfs(int x,int f,int id){
	fa[x][0]=f;
	costd[x]=costd[f]+val1[id];costu[x]=costu[f]+val2[id];
	num[x]=num[f]+a[x];dep[x]=dep[f]+1;
	for(int i=1;i<20;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
	dp[x]+=a[x];
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(y==f) continue;
		dfs(y,x,i);
		dp[x]+=c[y];
	}
	c[x]=max(0ll,dp[x]-val1[id]-val2[id]);
	return ;
} 
void dfs1(int x,int f,int id){
	c1[x]=c1[f]+c[x];dpp[x]=dpp[f]+dp[x];
	dp1[x]=max(dp1[f]+dp[f]-c[x]-val1[id]-val2[id],0ll);
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(y==f) continue;
		dfs1(y,x,i);
	}
//	cout<<x<<" costd:"<<costd[x]<<" costu:"<<costu[x]<<" num:"<<num[x]<<" c:"<<c[x]<<" dp:"<<dp[x]<<" dp1:"<<dp1[x]<<" dpp:"<<dpp[x]<<" c1:"<<c1[x]<<endl;
	
	return;
}
inline int Getlca(int u,int v){
	if(u==v) return u;
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=19;i>=0;i--) if(dep[fa[u][i]]>=dep[v]) u=fa[u][i];
	if(u==v) return u;
	for(int i=19;i>=0;i--) if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}
inline int Jump(int x,int lca){
	for(int i=19;i>=0;i--) if(dep[fa[x][i]]>dep[lca]) x=fa[x][i];
	return x;
}
signed main(){
//	freopen("strawberry.in","r",stdin);
//	freopen("strawberry.out","w",stdout);
	read(n);read(q);
	for(int i=1;i<=n;i++) read(a[i]);
	for(int i=1;i<n;i++){
		int u,v,w1,w2;
		read(u),read(v),read(w1),w2=w1;
		add(u,v,w1,w2);
		add(v,u,w2,w1);
	}
	dfs(1,0,0);c[1]=0;
	dfs1(1,0,0);
	while(q--){
		int x,y;
		read(x),read(y);
		int lca=Getlca(x,y),ls=Jump(x,lca),rs=Jump(y,lca);
//		cout<<"LCA:"<<lca<<" ls:"<<ls<<" rs:"<<rs<<endl;
		if(x==lca) ls=0;
		if(y==lca) rs=0;
		ll Ans=0;
		if(x==y) Ans=dp[x]+dp1[x];
		else Ans=dpp[x]+dpp[y]-2*dpp[lca]+dp[lca]+dp1[lca]-(c1[x]+c1[y]-c1[lca]*2)-costu[x]+costu[lca]-costd[y]+costd[lca];
//		else Ans=dp[lca]+dp1[lca]-c[ls]-c[rs]-costu[x]+costu[lca]-costd[y]+costd[lca]+num[x]+num[y]-2*num[lca];
		write(Ans),putchar('\n');
	}
	return 0;
}
上一篇:LeetCode-071-简化路径


下一篇:Paths on a Grid题解