题意:给你一颗树问你使所有路径的异或和都不为0的最小操作次数
这题可以用dsu on tree 或者 启发式合并 解决
解题思路:如果先把一个数修改成很大的一个数异或和就可以不为0,维护一个到根节点的异或和,每次看一条路径的异或和就是直接两个异或和在异或上两个的父亲,这个过程可以用dsu on tree或者启发式来解决,如果为0就要根节点就一定要修改,最后统计要修改的就为答案
启发式合并比较好实现
#include <bits/stdc++.h>
using namespace std;
#define x first
#define y second
#define ll long long
#define pii pair<int, int>
#define all(x) (x).begin(), (x).end()
const int N = 2e5 + 10;
int sz[N];
int fa[N];
int a[N], b[N];
int sum[N];
vector<int> h[N];
set<int> st[N];
int n, k;
int ans;
void dfs(int u, int f)
{
sum[u]=sum[f]^a[u];
st[u].insert(sum[u]);
for(auto v : h[u])
{
if(v == f) continue;
dfs(v, u);
if(st[v].size() > st[u].size()) swap(st[v], st[u]); // 启发式合并
for(auto i : st[v]) if(st[u].count(a[u]^i)) b[u] = 1; //如果找到为0就要删除
for(auto i : st[v]) st[u].insert(i);
}
if(b[u]) st[u].clear(); //删除这个子树
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >>n;
for(int i = 1; i <= n; ++i) cin >>a[i];
for(int i = 0; i < n - 1; ++i)
{
int a, b;
cin >>a >>b;
h[a].push_back(b), h[b].push_back(a);
}
dfs(1, 0);
for(int i = 1; i <= n; ++i) if(b[i]) ans++;
cout<<ans<<endl;
}
dsu on tree 每次保留重儿子,合并轻儿子基本上是每次都可以套模板,这题需要修改需要注意一些细节,如果当前要修改直接跳过这个子树,用dfs序可以简化很多操作
dsu on tree:
#include <bits/stdc++.h>
using namespace std;
#define x first
#define y second
#define ll long long
#define pii pair<int, int>
#define all(x) (x).begin(), (x).end()
const int N = 2e5 + 10;
int hs[N], l[N], r[N], id[N], tot;
int sz[N];
int fa[N];
int a[N], b[N];
int sum[N];
vector<int> h[N];
set<int> st;
int n, k;
int ans;
void dfs1(int u, int f)
{
l[u] = ++tot;
id[tot] = u;
sum[u]=sum[f]^a[u];
hs[u] = -1;
sz[u] = 1;
for(auto v : h[u])
{
if(v == f) continue;
dfs1(v, u);
sz[u]+=sz[v];
if(hs[u] == -1 || (sz[v] > sz[hs[u]])) hs[u] = v;
}
r[u] = tot;
}
void dfs2(int u, int f, bool ok)
{
for(auto v : h[u])
{
if(v == f || v == hs[u]) continue;
dfs2(v, u, 0);
}
if(hs[u]!=-1) dfs2(hs[u], u, 1);
bool flag = 0;
auto add = [&](int x)
{
st.insert(sum[x]);
};
auto query = [&](int x)
{
if(st.count(a[u]^sum[x])) flag = 1;
};
for(auto v : h[u])
{
if(v == f || v == hs[u]) continue;
if(flag) break;
for(int i = l[v]; i <= r[v]; ++i)
{
if(b[id[i]]) i = r[id[i]]; // 如果删除就直接跳过这个子树
else
query(id[i]);
}
for(int i = l[v]; i <= r[v]; ++i)
{
if(b[id[i]]) i = r[id[i]];
else
add(id[i]);
}
}
query(u), add(u);
if(flag) b[u] = 1, st.clear();
if(!ok) st.clear();
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >>n;
for(int i = 1; i <= n; ++i) cin >>a[i];
for(int i = 0; i < n - 1; ++i)
{
int a, b;
cin >>a >>b;
h[a].push_back(b), h[b].push_back(a);
}
dfs1(1, 0);
dfs2(1, 0, 0);
for(int i = 1; i <= n; ++i) if(b[i]) ans++;
cout<<ans<<endl;
}