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

这题只能分解轻重儿子吗,像下面这种暴力交换的一直过不去,是我有地方可以优化吗,求助

#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<pair<int, 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;
        g[x].emplace_back(i, 1 << (c - 'a'));
    }
    for (int i = 1; i <= n; i++) {
        id[i] = i;
    }
    function<void(int)> dfs = [&](int u) {
        mp[u][mask[u]] = dep[u];
        for (auto [v, c] : g[u]) {
            dep[v] = dep[u] + 1;
            mask[v] = mask[u] ^ c;
            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:38
加载中...