相关例题
食物链
D. The Number of Imposters
食物链
题目大意
给出一种食物链,其中A吃B,B吃C,C吃A,构成一种环形食物链
给出m个输入,其中包括两种输入k,x,y:
1、当k = 1时,x 和 y为同类
2、当k = 2时,表示 x 吃 y
这 m 个输入中有的是假话,当满足以下三个条件之一,即为假话
1、当前的话和以前的话存在冲突
2、x 或 y 的值比n大
3、x 吃 x
求这m个输入中有多少假话
方法一(带权并查集):
通过一个数组d来表示所有点到根节点的距离,默认为0,其中距离 %3 的三种结果0 , 1, 2表示三种动物
过题代码
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 5e4 + 10;
int d[N],p[N];
int find(int x)
{
if(x != p[x])
{
int t = find(p[x]);
d[x] += d[p[x]];
p[x] = t;
}
return p[x];
}
int main()
{
int n,m;
cin >> n >> m;
for(int i = 1;i <= n;i ++)
p[i] = i;
int ans = 0;
while (m -- )
{
int k,a,b;
cin >> k >> a >> b;
if(a > n || b > n)
{
ans ++ ;
continue;
}
int fa = find(a),fb = find(b);
if(k == 1)
{
if(fa == fb && (d[a] - d[b]) % 3)ans ++;//a 和 b不符合关系
else if(fa != fb){
p[fa] = fb;
d[fa] = d[b] - d[a];
}
}else {
if(fa == fb && (d[a] - d[b] - 1) % 3)ans ++;//a 和 b不符合关系
else if(fa != fb)
{
p[fa] = fb;
d[fa] = d[b] - d[a] + 1;
}
}
}
cout << ans << "\n";
return 0;
}
方法二(扩展域并查集)
创造3*n的数组空间,x,x + n,x + n + n分别表示x的同类、捕食域和天敌域
当k = 1时,p[x + n] 和 p[x + n + n] 中含有y为假话
当k = 2时,p[x] 和 p[x +n +n]中含有y为假话
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 5e4 + 10;
int p[3 * N];
int n,m,k,a,b;
int find(int x)
{
if(x != p[x])
p[x] = find(p[x]);
return p[x];
}
void me(int a,int b)
{
int fa = find(a),fb = find(b);
p[fa] = p[fb];
return;
}
int main()
{
cin >> n >> m;
int ans = 0;
for(int i = 1;i <= 3 * n;i ++)
p[i] = i;
while (m -- )
{
cin >> k >> a >> b;
if(a > n || b > n){
ans ++;
continue;
}
if(k == 1)
{
if(find(a + n) == find(b) || find(a + n + n) == find(b))ans ++;//a的捕食域或天敌是b
else {
me(a,b);
me(a + n,b + n);
me(a + n + n,b + n + n);
}
}else {
if(find(a) == find(b) || find(a + n + n) == find(b))ans ++;//a的同类或者天敌是b
else {
me(a,b + n + n);
me(a + n,b);
me(a + n + n, b + n);
}
}
}
cout << ans << "\n";
return 0;
}
D. The Number of Imposters(cf 1594D)
题目大意
与食物链题类似,对于x,y,ch 共有两种输入方式
1、当ch = imposter 时,表示 x 说 y 是坏人
2、当ch = crewmate时,表示 x 说 y 是好人
并且好人说的都是真话,坏人说的都是假话
题目分析
通过分析容易发现,当 x 说 y 是好人时,不论 x 是不是好人,x 和 y 都是一伙的;当 x 说 y 是坏人时,不论 x 是不是好人,x 和 y都是对立的
于是,我们可以用上面两种方法解决这道问题
方法一(带权并查集)
解题代码
//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=2e18+100;
const int maxn=2e5+100;
int a[maxn],pre[maxn],cnt[maxn][2];
int n,m;
int find(int x)
{
if(x!=pre[x])
{
int root=find(pre[x]);
a[x]^=a[pre[x]];
pre[x]=root;
}
return pre[x];
}
signed main()
{
IOS
int tt;
cin>>tt;
while(tt--)
{
bool fg=1;
cin>>n>>m;
for(int i=1; i<=n; i++)
{
pre[i]=i;
a[i]=0;
cnt[i][0]=1;
cnt[i][1]=0;
}
for(int i=1; i<=m; i++)
{
int x,y;
string str;
cin>>x>>y>>str;
int val=0;
if(str[0]=='i')val=1;
int fx=find(x),fy=find(y);
if(fx==fy)
{
if((a[x]^a[y])!=val)fg=0;
}
else
{
pre[fy]=fx;
if((a[x]^a[y])!=val)
{
a[fy]^=1;
cnt[fx][1]+=cnt[fy][0];
cnt[fx][0]+=cnt[fy][1];
}
else
{
cnt[fx][1]+=cnt[fy][1];
cnt[fx][0]+=cnt[fy][0];
}
}
}
int ans=0;
for(int i=1;i<=n;i++)
{
if(pre[i]==i)ans+=max(cnt[i][0],cnt[i][1]);
}
if(fg==0)cout<<"-1"<<"\n";
else cout<<ans<<"\n";
}
}
方法二(扩展域并查集)
解题代码
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <stack>
#include <queue>
#include <cmath>
#include <cstring>
#include <set>
#include <list>
#include <unordered_set>
#include <unordered_map>
#define gcd __gcd
#define LL long long
#define PII pair<int,int>
#define MII map<int,int>
#define MIS map<int,string>
#define MSI map<string,int>
#define VI vector<int>
#define pb push_back
#define fi first
#define se second
#define all(x) (x).begin(),(x).end()
using namespace std;
const int N = 2e5 + 10,M = 1e5 + 10;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
int p[2 * N],res[2 * N];
int n,m,a,b;
string s;
int find(int x)
{
if(x != p[x])p[x] = find(p[x]);
return p[x];
}
void me(int a,int b)
{
int fa = find(a),fb = find(b);
if(fa != fb)
{
p[fa] = fb;
res[fb] += res[fa];
res[fa] = 0;
}
}
int main()
{
int tt;
cin >> tt;
while(tt--)
{
cin >> n >> m;
bool flag = true;
for(int i = 1;i <= n;i ++)
p[i] = i,res[i] = 1,res[i + n] = 0,p[i + n] = i + n;
while(m--)
{
cin >> a >> b >> s;
if(s[0] == 'i')
{
if((find(a) == find(b) || find(a + n) == find(b + n)))flag = false;
else {
me(a,b + n);
me(a + n,b);
}
}else {
if((find(a) == find(b + n)|| find(a + n) == find(b)))flag = false;
else {
me(a , b);
me(a + n,b + n);
}
}
}
if(flag){
int ans = 0;
for(int i = 1;i <= n;i ++)
ans += max(res[i],res[i + n]);
cout << ans << "\n";
}else cout << "-1\n";
}
return 0;
}