题目描述
题解
这道题 O ( n 2 ) O(n^2) O(n2)很简单,就是一个简单的有根树树形DP,即经过每一个特殊点的边权*2-根到某一个特殊点的最长链。
对于部分1,我们假设处理了 f [ x ] f[x] f[x],对于一条边 ( x , y , v ) (x,y,v) (x,y,v)来说,在一般情况下有: f [ y ] = f [ x ] f[y]=f[x] f[y]=f[x]
- 若在以y为跟的情况下可以不到达x,也就是 s i z e [ y ] = k size[y]=k size[y]=k,那么我们可以减去边权。
- 如果这条边原来没有被累加过,也就是 s i z e [ y ] = 0 size[y]=0 size[y]=0,我们可以加上边权。
对于以某一个为根的最长链来说,我们对于边 ( x , y , v ) (x,y,v) (x,y,v)来说,求出 x x x的儿子深度的最大值和次大值。
- 如果 d e p [ y ] + v = max 1 dep[y]+v=\max_1 dep[y]+v=max1,用 v + max 2 v+\max_2 v+max2更新答案。
- 否则用 v + max 1 v+\max_1 v+max1更新答案即可。
要注重程序的实现方式。
代码如下:
#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;
}