Educational Codeforces Round 132 (Rated for Div. 2) E

题意:给你一颗树问你使所有路径的异或和都不为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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值