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