G-多吃蘑菇（树形dp）

G-多吃蘑菇

题意：

给定一棵树，树上的节点有颜色与权值，求一条简单路径，路径上的权值和最大，同颜色只能统计一次。

做法：

从根节点开始往下遍历，同时维护一个数组，表示从当前节点到根的路径上，出现过的颜色的最大值。很明显节点可能会更新这个数组，更新完后，可以继续往下dfs，当以为根的子树遍历完后，对数组的贡献可以取消了就。

代码：（比赛时的代码，用了map来代替）

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e5 + 5, mod = 1e9 + 7;
int c[N], w[N];
vector<int> v[N];
map<int, int> f[N];
int ans[N];
void dfs(int x, int fa)
{
    for (int i : v[x])
    {
        int y = i;
        if (y == fa) continue;
        swap(f[y], f[x]);
        int temp = -1;
        if (f[y].count(c[y])) temp = f[y][c[y]];
        ans[y] = ans[x];
        ans[y] += max(0ll, w[y] - f[y][c[y]]);
        f[y][c[y]] = max(f[y][c[y]], w[y]);
        dfs(y, x);
        swap(f[x], f[y]);
        if (temp == -1) f[x].erase(c[y]);
        else f[x][c[y]] = temp;
    }
}
signed main()
{
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) scanf("%lld", &w[i]);
    for (int i = 1; i <= n; i++) scanf("%lld", &c[i]);
    for (int i = 1; i <= n - 1; i++)
    {
        int x, y;
        scanf("%lld%lld", &x, &y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    f[1][c[1]] = w[1];
    ans[1] = w[1];
    dfs(1, 0);
    for (int i = 1; i <= n; i++) printf("%lld\n", ans[i]); 
    return 0;
}