『树形DP·换根』Kamp

题目描述

题解

这道题$O(n^2)$很简单，就是一个简单的有根树树形DP，即经过每一个特殊点的边权*2-根到某一个特殊点的最长链。

对于部分1，我们假设处理了$f[x]$,对于一条边$(x,y,v)$来说，在一般情况下有：$f[y]=f[x]$

• 若在以y为跟的情况下可以不到达x，也就是$size[y]=k$,那么我们可以减去边权。
• 如果这条边原来没有被累加过，也就是$size[y]=0$,我们可以加上边权。

对于以某一个为根的最长链来说，我们对于边$(x,y,v)$来说，求出$x$的儿子深度的最大值和次大值。

• 如果$dep[y]+v={\max }_1$,用$v+{\max }_2$更新答案。
• 否则用$v+{\max }_1$更新答案即可。

要注重程序的实现方式。

代码如下：

#include <bits/stdc++.h>
#define int long long

using namespace std;
const int N = 600000;

int n, m;
int size[N], vis[N], f[N], g[N], s[N];

vector < pair<int,int> > a[N];

bool dfs(int x,int fa)
{
	size[x] = vis[x];
	for (int i=0;i<a[x].size();++i)
	{
		int y = a[x][i].first;
		int v = a[x][i].second;
		if (y == fa) continue;
		if (dfs(y,x)) 
		{
			f[x] = max(f[x],f[y]+v);
			s[x] += s[y]+v*2;
			size[x] += size[y];
		}
	}
	return vis[x] || f[x];
}

void dp(int x,int fa)
{
	int Max1 = 0, Max2 = 0;
	for (int i=0;i<a[x].size();++i)
	{
		int y = a[x][i].first;
		int v = a[x][i].second;
		if (y == fa) continue;
		if (!vis[y] && !f[y]) continue;
		int s[3] = {f[y]+v,Max1,Max2};
		sort(s,s+3); Max1 = s[2], Max2 = s[1];
	}
	for (int i=0;i<a[x].size();++i)
	{
		int y = a[x][i].first;
		int v = a[x][i].second;
		if (y == fa) continue;
		s[y] = s[x];
		if (size[y] == m) s[y] -= v * 2;
		if (size[y] == 0) s[y] += v * 2;
		if (g[x] > 0) g[y] = g[x]+v;
		if (f[y] + v == Max1 && Max2 != 0) g[y] = max(g[y],Max2+v);
		if (f[y] + v != Max1 && Max1 != 0) g[y] = max(g[y],Max1+v);
		dp(y,x);
	}
	return;
}

signed main(void)
{
	freopen("kamp.in","r",stdin);
	freopen("kamp.out","w",stdout);
	scanf("%lld %lld", &n, &m);
	for (int i=1,x,y,v;i<n;++i)
	{
		scanf("%lld %lld %lld", &x, &y, &v);
		a[x].push_back(make_pair(y,v));
		a[y].push_back(make_pair(x,v));
	}
	for (int i=1,x;i<=m;++i)
		scanf("%lld", &x), vis[x] = 1;
	dfs(1,0);
	dp(1,0);
	for (int i=1;i<=n;++i)
		printf("%lld\n", s[i]-max(g[i],f[i]));
	return 0;
}