暴力求助
查看原帖
暴力求助
702623
Yakult楼主2024/4/3 19:28

这题只能轻重儿子分解吗,我下面写的暴力交换,一直过不去,被卡爆了。还是有能优化的地方呢,求大佬指点

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
const int N = 5e5 + 10;
map<int, int> mp[N];
vector<int> g[N];
int ans[N], dep[N], mask[N], id[N];
void solve() {
    int n;
    cin >> n;
    for (int i = 2; i <= n; i++) {
        int x;
        char c;
        cin >> x >> c;
        dep[i] = dep[x] + 1;
        mask[i] = mask[x] ^ (1 << (c - 'a'));
        g[x].emplace_back(i);
    }
    for (int i = 1; i <= n; i++) id[i] = i;
    function<void(int)> dfs = [&](int u) {
        mp[u][mask[u]] = dep[u];
        for (int v : g[u]) {
            dfs(v);
            ans[u] = max(ans[u], ans[v]);
            if (mp[id[u]].size() < mp[id[v]].size()) {
                swap(id[u], id[v]);
            }
            int res = 0;
            for (auto &[k, v] : mp[id[v]]) {
                if (mp[id[u]].count(k)) {
                    res = max(res, v + mp[id[u]][k]);
                }
                for (int j = 0; j < 22; j++) {
                    if (mp[id[u]].count(k ^ (1 << j))) {
                        res = max(res, v + mp[id[u]][k ^ (1 << j)]);
                    }
                }
            }
            for (auto &[k, v] : mp[id[v]]) {
                mp[id[u]][k] = max(mp[id[u]][k], v);
            }
            ans[u] = max(ans[u], res - 2 * dep[u]);
        }
    };
    dfs(1);
    for (int i = 1; i <= n; i++) {
        cout << ans[i] << " ";
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int T = 1;
    while (T--) {
        solve();
    }
    return 0;
}

2024/4/3 19:28
加载中...