C. Parsa’s Humongous Tree
哈哈 树状数组。
题目传送门:
题面:
中文题意:
意思是说,有个树,每个节点的值可以从 l , r l,r l,r中选择一个,每个边 ( u , v ) (u,v) (u,v)的权值为 ( a i − a j ) (a_i-a_j) (ai−aj),要求整棵树边权值总和最大。
思路:
挺明显的树形DP裸题。
d p [ f a ] [ 0 ] + = m a x ( d p [ s o n ] [ 0 ] + ∣ l s o n − l f a ∣ , d p [ s o n ] [ 1 ] + ∣ r s o n − l f a ∣ dp[fa][0]+=max(dp[son][0]+|lson-l_fa|,dp[son][1]+|rson-l_fa| dp[fa][0]+=max(dp[son][0]+∣lson−lfa∣,dp[son][1]+∣rson−lfa∣
其实很好理解的,意思是说 d p [ f a ] [ 0 ] dp[fa][0] dp[fa][0]就是fa这个节点采用 l f a l_fa lfa这个值,他造成的影响就是从上一个son节点转移过来的时候,需要运算的就是 l f a l_fa lfa,根据son节点取值的不同情况选择最优的进行转移。
d p [ f a ] [ 1 ] + = m a x ( d p [ s o n ] [ 0 ] + ∣ l s o n − r f a ∣ , d p [ s o n ] [ 1 ] + ∣ r s o n − r f a ∣ dp[fa][1]+=max(dp[son][0]+|lson-r_fa|,dp[son][1]+|rson-r_fa| dp[fa][1]+=max(dp[son][0]+∣lson−rfa∣,dp[son][1]+∣rson−rfa∣
同上。
那么实现过程应该是这样:
- 写一个dfs填充dp数组。
- main函数建图,然后跑dfs,输出根节点的dp两个最大值。
代码:
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
const int maxn = 114514;
ll l[maxn], r[maxn];
vector<ll> v1[maxn];
ll dp[maxn][3];
void dfs(ll x, ll y) {
dp[x][1]=0;
dp[x][0]=0;
//记得变成0
for (ll i = 0; i < v1[x].size(); i++) {
ll son = v1[x][i];
if (son == y)
continue;
//跳出条件;
dfs(son, x);
//递归,得到父节点dp先要算完叶结点。
dp[x][1] += max(dp[son][0] + abs(l[son] - r[x]), dp[son][1] + abs(r[son] - r[x]));
dp[x][0] += max(dp[son][0] + abs(l[son] - l[x]), dp[son][1] + abs(r[son] - l[x]));
//dp含义前面说了。
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
int t;
cin >> t;
while (t--) {
ll n;
cin>>n;
for (int i = 1; i <= n; i++)
{
cin >> l[i] >> r[i];
v1[i].clear();
//记得clear()
}
for (int i = 1; i < n ; i++) {
ll u, v;
cin >> u >> v;
v1[u].push_back(v);
v1[v].push_back(u);
}
dfs(1, 0);
//从根节点开始往下面深搜,边界值为0
cout << max(dp[1][1], dp[1][0]) << endl;
}
}