Codeforces Round #722 (Div. 2) C. Parsa‘s Humongous Tree（树形DP）

本文链接：https://blog.csdn.net/qq_27851423/article/details/117733596

题意

给你一棵树，树上的每一个点都有一个值区间，让你求树上相邻各点的值之差的最大值

输入

前n行是1~n的各个点的值区间
之后n-1行表示哪两个点之间是链接的

思路

首先，区间是没有用的，只可能取端点的值；
以下为证明：
设区间为l，r;
不取端点l,r，任取一点mid，使得l<mid<r;
再设在区间 [l,r] 左端有p个点，右段有q个点，
令p>q；显而易见用r代替mid时各点差的和更大，
p<q时同理可证，所以我使用 dp[x][2]来表示x为顶点取左右端点的最大值
给出状态转移方程：

        dp[x][0] += 1ll * max(dp[y][0] + abs(l[x] - l[y]), dp[y][1] + abs(l[x] - r[y]));
        dp[x][1] += 1ll * max(dp[y][0] + abs(r[x] - l[y]), dp[y][1] + abs(r[x] - r[y]));

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll tot=0,head[400007],ver[800007],nextt[800007],l[400007],r[400007],dp[400007][2];
void add(int x,int y)//加点
{
    tot++;
    ver[tot]=y;
    nextt[tot]=head[x];
    head[x]=tot;
}
void dfs(int x,int pre)//dfs
{
    for(int i=head[x]; i; i=nextt[i])
    {
        int y=ver[i];
        if(y==pre)
            continue;
        dfs(y,x);
        dp[x][0] += 1ll * max(dp[y][0] + abs(l[x] - l[y]), dp[y][1] + abs(l[x] - r[y]));
        dp[x][1] += 1ll * max(dp[y][0] + abs(r[x] - l[y]), dp[y][1] + abs(r[x] - r[y]));
    }
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
    	tot=0;
        int n;
        cin>>n;
        for(int i=1; i<=n; i++)
        {
            scanf("%lld %lld",&l[i],&r[i]);
            dp[i][0]=0;
            dp[i][1]=0;
            head[i]=0;
        }
        for(int i=1;i<=n-1;i++)
        {
        	int p,q;
        	scanf("%d %d",&p,&q);
        	add(p,q);
        	add(q,p);//正反加
		}
		dfs(1,0);
		cout<<max(dp[1][0],dp[1][1])<<endl;
    }
}