算法竞赛进阶指南 0X40数据结构进阶——奇偶游戏

题目链接

AcWing 239. 奇偶游戏 mid

题目描述

小 $A$ 和小 $B$ 在玩一个游戏。

首先，小 $A$ 写了一个由 $0$ 和 $1$ 组成的序列 $S$，长度为 $N$。

然后，小 $B$ 向小 $A$ 提出了 $M$ 个问题。

在每个问题中，小 $B$ 指定两个数 $l$ 和 $r$，小 $A$ 回答 $S[l\sim r]$ 中有奇数个 $1$ 还是偶数个 $1$。

机智的小 $B$ 发现小 $A$ 有可能在撒谎。

例如，小 $A$ 曾经回答过 $S[1\sim 3]$ 中有奇数个 $1$，$S[4\sim 6]$ 中有偶数个 $1$，现在又回答 $S[1\sim 6]$ 中有偶数个 $1$，显然这是自相矛盾的。

请你帮助小 $B$ 检查这 $M$ 个答案，并指出在至少多少个回答之后可以确定小 $A$ 一定在撒谎。

即求出一个最小的 $k$，使得 $01$ 序列 $S$ 满足第 $1\sim k$个回答，但不满足第 $1\sim k+1$ 个回答。

输入格式

第一行包含一个整数 $N$，表示 $01$ 序列长度。

第二行包含一个整数 $M$，表示问题数量。

接下来 $M$ 行，每行包含一组问答：两个整数 $l$ 和 $r$，以及回答 $even$ 或 $odd$，用以描述 $S[l\sim r]$ 中有偶数个 $1$ 还是奇数个 $1$。

输出格式

输出一个整数 $k$，表示 $01$ 序列满足第 $1\sim k$ 个回答，但不满足第 $1\sim k+1$ 个回答，如果 $01$ 序列满足所有回答，则输出问题总数量。

数据范围

$N\le 10^9,M\le 5000$

输入样例：

10
5
1 2 even
3 4 odd
5 6 even
1 6 even
7 10 odd

输出样例：

3

解法：带权值的并查集

如果我们使用 $sum$ 来表示序列 $S$ 的前缀和的话。

• $S[l:r]$有偶数个 $1$，说明 $sum[l-1]$ 和 $sum[r]$ 奇偶性相同；
• $S[l:r]$有奇数个 $1$，说明 $sum[l-1]$ 和 $sum[r]$ 奇偶性不同；

传递关系：

• 如果 $a$ 和 $b$ 奇偶性相同，$b$ 和 $c$ 奇偶性相同，那么 $a$ 和 $c$ 奇偶性也相同；
• 如果 $a$ 和 $b$ 奇偶性相同，$b$ 和 $c$ 奇偶性不同，那么 $a$ 和 $c$ 奇偶性也不同；
• 如果 $a$ 和 $b$ 奇偶性不同，$b$ 和 $c$ 奇偶性不同，那么 $a$ 和 $c$ 奇偶性也相同；

因为 $N$ 很大，而 $M$ 比较小，所以我们还是使用离散化的技巧，将输入的数映射到 最大为 $2M$ 的数组中。在实现上，我们可以使用 map 来实现。

我们定义 $d[x]$ 为 $x$ 节点到它的父结点 $p[x]$ 的边权。

• 如果 $d[x]=0$，表示节点 $x$ 与它的父节点 $p[x]$ 奇偶性相同；
• 如果 $d[x]=1$，表示节点 $x$ 与它的父节点 $p[x]$ 奇偶性不同；

路径压缩：

在路径压缩的时候，对于 节点$x$ 到其根节点路径上的所有边权值做 异或运算，就可以得到 节点$x$ 与其 根节点的奇偶性关系。

对于 $x$ 和 $y$ ，我们先检查它们是否在一个集合中，即 $find(x)=find(y)$。

1. 如果在同一个集合中。如果 $d[x]xord[y]\ne type$，说明这个回答有矛盾，则在这个问题之后就能够确定小 $A$ 撒谎了。
2. 如果不在同一个集合之中，就需要先合并两个集合。

我们设 $p,q$ 分别是 $x,y$ 的祖宗节点。

我们已知 $d[x]$ 和 $d[y]$ 分别表示 路径 $x\sim p$ 和 $y\sim p$ 之间的所有边权 异或和。

$p\sim q$ 这一段路径的异或和 $d[p]$ 是我们要求的 。

由于 $x\sim y$ 的路径是由 $x\sim p,p\sim q,q\sim y$ 组成的，$x$ 和 $y$ 的奇偶性 $type$ 我们是已知的，因此 $type=d[x]xord[y]xord[p]$。

所以 $d[p]=d[x]xord[y]xortype$。

时间复杂度： $O(n)$

C++代码：

#include<iostream>
#include<algorithm>
#include<map>

using namespace std;

const int N = 1e4+10;

struct{
    int l,r,type;
}query[N];

int n,m;

int p[N] , d[N] , idx = 0;

map<int,int> mp;

//输入离散化
void read_discrete(){
    cin>>n>>m;
    char str[5];
    
    for(int i = 1;i <= m;i++){
        
        scanf("%d%d%s",&query[i].l,&query[i].r,str);
        
        query[i].type = (str[0] == 'o' ? 1 : 0);
        
        int l = query[i].l - 1 , r = query[i].r;
        
        if(mp.find(l) == mp.end()) mp[l] = ++idx;
        if(mp.find(r) == mp.end()) mp[r] = ++idx;
    }
    
    n = idx;
}

//路径压缩找根节点
int find(int x){
    if(x == p[x]) return p[x];
    
    int root = find(p[x]);
    d[x] ^= d[p[x]];

    return p[x] = root;
}

int main(){
    read_discrete();
    
    //cout<<n<<'\n';
    
    for(int i = 1;i <= n;i++) p[i] = i;
    
    for(int i = 1;i <= m;i++){
        
        int l = query[i].l - 1 , r = query[i].r;
        
        int a = mp[l] , b = mp[r];
        
        int x = find(a) , y = find(b);
        
        if(x == y){
            //i轮发生矛盾,i - 1 轮及之前的都是没问题的 
            if((d[a] ^ d[b]) != query[i].type){
                cout<<i - 1<<'\n';
                return 0;
            }
        }
        
        else{
            p[x] = y;
            d[x] = d[a] ^ d[b] ^ query[i].type;
        }
    }
    
    //没有矛盾
    cout<<m<<'\n';
    return 0;
}