bytecamp

 #include<bits/stdc++.h>

using namespace std;
const long long mod = 1e9 + 7;

const int maxn = 20005;

struct node {
    int v;
    char w;
};
vector<node> G[maxn];
unordered_map<unsigned long long, int> mp[maxn];
int ans[maxn];
int n, q, u, v, st, val[maxn];
char w;

void dfs(unsigned long long e, int x, int fa) {
    mp[x][e] = val[x];
    if (G[x].size() == 1 && G[x][0].v == fa) {
        return;
    }
    for (int i = 0; i < G[x].size(); ++i) {
        if (G[x][i].v == fa) {
            continue;
        }
        dfs(e * 131 + (G[x][i].w - 'a' + 1), G[x][i].v, x);
        for (auto &k:mp[G[x][i].v]) {
            mp[x][k.first] += k.second;
        }
    }
    for (auto &i:mp[x]) {
        ans[x] = max(ans[x], i.second);
    }
    mp[x].clear();
}

int main() {
    ios::sync_with_stdio(0);
    cin >> n >> q;
    for (int i = 1; i <= n; ++i) {
        cin >> val[i];
    }
    for (int i = 1; i < n; ++i) {
        cin >> u >> v >> w;
        G[u].push_back({v, w});
        G[v].push_back({u, w});
    }
    dfs(0, 1, 1);
    while (q--) {
        cin >> st;
        cout << ans[st] << '\n';
    }
    return 0;
}


在这里插入图片描述

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值