牛客月赛22-树上子链-(树的直径点权和边权)

B

题意：
就是给你一个数，每个节点有一个权值，现在问你一条链的所有节点权值之和最大为多少。

思考：
其实这就是树的直径的做法，但是这个呢是点权不是边权，如果是边权那么用两次dfs或者树上dp。两次dfs的做法还能求出树的直径的两个端点，但是不能做负边权的。这个题就是树上dp就行了。

代码：

int T,n,m,k;
int maxn = -inf;
int va[N];
int dp[N];

vector<int > e[N];

void dfs(int now,int p)
{
	dp[now] = va[now];
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		dfs(spot,now);
        maxn = max(maxn,dp[now]+dp[spot]); //now走的链和另一条链加起来
		dp[now] = max(dp[now],va[now]+dp[spot]); //now到底走哪条链
	}
	maxn = max(maxn,dp[now]); //如果只走一条链
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	dfs(1,0);
	cout<<maxn;
	return 0;
}

维护最大值和次大值做法(和求边权的一样)：
int T,n,m,k;
int maxn = -inf;
int va[N];
int dp[N][2];

vector<int > e[N];

void dfs(int now,int p)
{
	dp[now][0] = dp[now][1] = va[now];
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		dfs(spot,now);
		if(dp[now][0]<=dp[spot][0]+va[now])
		{
			dp[now][1] = dp[now][0];
			dp[now][0] = dp[spot][0]+va[now];
		}
		else if(dp[spot][0]+va[now]>dp[now][1]) dp[now][1] = dp[spot][0]+va[now];
	}
	maxn = max(maxn,dp[now][0]+dp[now][1]-va[now]); //减去是因为次大值也选了当前点，要去掉。
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	dfs(1,0);
	cout<<maxn;
	return 0;
}

求树的直径(边权)：

int T,n,m,k;
int maxn;
in
t va[N];
int dp[N][2];

vector<PII > e[N];

void dfs(int now,int p)
{
	for(auto t:e[now])
	{
		int spot = t.fi,w = t.se;
		if(spot==p) continue;
		dfs(spot,now);
		if(dp[now][0]<dp[spot][0]+w)
		{
			dp[now][1] = dp[now][0];
			dp[now][0] = dp[spot][0]+w;
		}
		else if(dp[spot][0]+w>dp[now][1]) dp[now][1] = dp[spot][0]+w;
	}
	maxn = max(maxn,dp[now][0]+dp[now][1]);
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		e[a].pb({b,c});
		e[b].pb({a,c});
	}
	dfs(1,0);
	cout<<maxn;
	return 0;
}

总结：
多多思考，弥补弥补弱项。