Codeforces 960E 树dp

E. Alternating Tree

题意：
给定一棵树，每个点有点权，求树上所有路径的价值总和。 一条 u1 -> u2 -> u3 -> .......-> um 路径的价值定义为 A(u1, um) += ((-1)^(i+1)) * val[i] ， (1<=i<=m) 。
tags：
树dp ，跑两遍 dfs ，求出 dp1[i][0] 和 dp1[i][1] 代表 i 结点上方到 i 点长度为奇和偶的个数， dp2[i][0] 和 dp2[i][1] 代表 i 结点下方到 i 点 长度为奇和偶的个数。
然后以每条边为基础，对每个点求贡献。

//   CF 960E
#include<bits/stdc++.h>
using namespace std;
#pragma comment(linker, "/STACK:102400000,102400000")
#define rep(i,a,b) for (int i=a; i<=b; ++i)
#define per(i,b,a) for (int i=b; i>=a; --i)
#define mes(a,b)  memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define MP make_pair
#define PB push_back
#define fi  first
#define se  second
typedef long long ll;
const int N = 200005, mod = 1e9+7;

int n;
ll  val[N], dp1[N][2], dp2[N][2];
vector< int > G[N];
void dfs1(int u, int fa)
{
    dp2[u][0] = 1;
    for(int to : G[u]) if(to!=fa) {
        dfs1(to, u);
        dp2[u][0] += dp2[to][1];
        dp2[u][1] += dp2[to][0];
    }
}
ll  ans = 0;
void dfs2(int u, int fa)
{
    ans = (ans + val[u]*n%mod + mod)%mod;
    for(int to : G[u]) if(to!=fa) {
        dp1[to][0] = dp2[u][1] - dp2[to][0] + dp1[u][1];
        dp1[to][1] = dp2[u][0] - dp2[to][1] + dp1[u][0];
        ans = (ans - val[to]*dp1[to][1]%mod*(dp2[to][0]+dp2[to][1])%mod+mod)%mod;
        ans = (ans + val[to]*dp1[to][0]%mod*(dp2[to][0]+dp2[to][1])%mod+mod)%mod;
        ans = (ans + val[u]*dp2[to][1]%mod*(dp1[to][0]+dp1[to][1])%mod+mod)%mod;
        ans = (ans - val[u]*dp2[to][0]%mod*(dp1[to][0]+dp1[to][1])%mod+mod)%mod;
        dfs2(to, u);
    }
}
int main()
{
    scanf("%d", &n);
    rep(i,1,n) scanf("%lld", &val[i]);
    int u, v;
    rep(i,1,n-1) {
        scanf("%d%d", &u, &v);
        G[u].PB(v),  G[v].PB(u);
    }
    dfs1(1, 0);
    dfs2(1, 0);
    printf("%lld\n", (ans+mod)%mod);

    return 0;
}

转载于:https://www.cnblogs.com/sbfhy/p/8819507.html