AcWing 239. 奇偶游戏—带边权并查集、扩展域
- 问题
- 带边权并查集解法
- 扩展域解法
- 并查集所要掌握的知识技能如下图所示
问题
题目链接: AcWing 239. 奇偶游戏
问题描述
分析
这道题比较有意思,可以由前缀和的思想来解决,[l,r]为偶数,说明[0,l-1]和[0,r]的奇偶性相同;[l,r]为奇数,说明[0,l-1]和[0,r]的奇偶性不同;
带边权并查集解法
这道题的解法类似于食物链 是一种带边权的并查集,维护当前节点相对父节点的距离即可,这个距离就表示了当前节点与父节点的关系,食物链那道题有三种情况,同类、吃、被吃,这道题只有两种情况,奇、偶
代码如下
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<unordered_map>
#include<string>
using namespace std;
const int N=1e4+10;
int f[N],idx=0,t[N];
unordered_map<int,int> mp;
int get_x(int x){
if(mp[x]==0) mp[x]=++idx;
return mp[x];
}
int find(int x){
if(x==f[x]) return x;
int d=f[x];
f[x]=find(f[x]);
t[x]+=t[d];
return f[x];
}
int main(){
int n,m,res=0;
cin>>n>>m;
for(int i=0;i<N;i++) f[i]=i;
while(m--){
int l,r;
string s;
cin>>l>>r>>s;
l--;
int x=get_x(l);
int y=get_x(r);
int fx=find(x);
int fy=find(y);
if(s=="even"){
if(fx==fy){
if((t[x]^t[y])%2==0) res++;
else break;
}
else{
res++;
f[fx]=fy;
t[fx]=(t[x]^t[y])%2;
}
}
else{
if(fx==fy){
if((t[x]^t[y])%2==1) res++;
else break;
}
else{
res++;
f[fx]=fy;
t[fx]=1-(t[x]^t[y])%2;
}
}
}
cout<<res;
return 0;
}
扩展域解法
扩展域的解法我是第一次遇到,大致的思想就是用并查集维护条件,当其中一个条件成立,那么集合中所有条件都应该成立。扩展域就是扩展出新的范围来枚举所有不同的情况。
举个例子,假设题目中x,y的范围<n,
1.如果说x,y是同类,那么
f[find(x)]=find(y)假设x,y都是偶数
f[find(x+n)]=find(y+n),假设x,y都是奇数
有这两种可能
判断是否违规,只需要判断x,y是否一个奇数、一个偶数即可
也就是
if(f[find(x+n)]==find(y))
或者
if(f[find(x)]==find(y+n))
2.如果说x,y是异类,那么
f[find(x)]=find(y+n) x是偶数,y是奇数
f[find(x+n)]=find(y) x是奇数,y是偶数
这两种情况
判断是否违规,只需要判断x,y是同为奇数或者同为偶数即可
也就是
if(f[find(x)]==find(y))
或者
if(f[find(x+n)]==find(y+n))
原数据的范围是[0,n],开辟一个新的空间(域)来表示不同的情况
代码如下
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<unordered_map>
#include<string>
using namespace std;
const int N=2e4+10,base=N>>1;
int f[N],idx=0,t[N];
unordered_map<int,int> mp;
int get_x(int x){
if(mp[x]==0) mp[x]=++idx;
return mp[x];
}
int find(int x){
if(x==f[x]) return x;
return f[x]=find(f[x]);
}
int main(){
int n,m,res;
cin>>n>>m;
for(int i=0;i<N;i++) f[i]=i;
res=m;
for(int i=1;i<=m;i++){
int l,r;
string s;
cin>>l>>r>>s;
l--;
int x=get_x(l);
int y=get_x(r);
if(s=="even"){
if(find(x+base)==find(y)){
res=i-1;
break;
}
f[find(x+base)]=find(y+base);
f[find(x)]=find(y);
}
else{
if(find(x)==find(y)){
res=i-1;
break;
}
f[find(x+base)]=find(y);
f[find(x)]=find(y+base);
}
}
cout<<res;
return 0;
}