Task200204.3 P5543 The Great Revegetate

Task2004.3 P5543 The Great Revegetation

题目描述:

一场漫长的干旱使农场主约翰的N牧场没有草, 然而, 随着雨季的到来, 是时候“重新种植”了。在农夫约翰的小屋里, 他有两个桶, 分别有一种草籽。他想在他的每一个牧场种草, 在每一个牧场中只能种一种草。 作为一名奶农, 农场主约翰想确保他能满足他的M头奶牛的特殊饮食需求。他每头奶牛都有两个自己最喜欢的牧场。一些奶牛只吃一种草, 因此约翰希望确保在这些奶牛最喜欢的两个牧场里种植同一种的草; 剩下的奶牛则吃不同种类的草。对于那些奶牛,约翰当然想确保它们最喜欢的两块牧场里有不同的草。 请帮助农场主约翰确定他在他的N块牧场上有多少种种植草的不同方式。

思路:

与关押罪犯有很多相似之处,实用并查集补集的思路,将两个牧场种相同草的关系当作「友好」,将两个牧场种不同草的关系当作「敌对」,但值得注意的一点是并非所有的牧场一定为任意一只奶牛所喜爱,所以我们还需要额外的一个并查集来维护被提到的牧场的连通性。

如果产生冲突,那么最终方案数必定为0;若没有冲突,就在第三个并查集中判断连通块,连通块数为为cnt,则$ans=2^{cnt}$;

代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,a,b,ans;
char op;
int fa[100005],gr[200005];
int find(int x){
    if(fa[x]==x) return x;
    else return fa[x]=find(fa[x]);
}
int com(int x){
    if(gr[x]==x) return x;
    else return gr[x]=com(gr[x]);
}
int main(){
    scanf("%d%d",&n,&m);
    ans=n;
    for(int i=1;i<=n;i++){
        gr[i]=i;
        gr[i+n]=i+n;
        fa[i]=i;
    }
    for(int i=1;i<=m;i++){
        cin>>op>>a>>b;
        // op=getchar();
        // getchar();
        // scanf("%d%d",&a,&b);
      //乖乖用cin(笑)
        if(op=='S'){
            gr[com(a)]=com(b);
            gr[com(a+n)]=com(b+n);
        }
        else{
            if(com(a)==com(b)){
                ans=0;
                break;
            }
            gr[com(a+n)]=com(b);
            gr[com(a)]=com(b+n);
        }
        if(find(a)!=find(b)){
            fa[find(a)]=find(b);
            ans--;
        }
    }
    if(ans==0){
        printf("%d",ans);
        return 0;
    }
    printf("1");
    for(int i=1;i<=ans;i++) printf("0");
    return 0;
}
Sorry potatoler, but I cannot recognize you.
Built with Hugo
Theme Stack designed by Jimmy