TopCoder SRM 618 Div1 easy Family

概要

N人に対し,その両親の情報が与えられる.両親はそれぞれ性別が異なる必要があるが,そのような組み合わせはあるか

解法

親同士を結んだグラフが二部グラフかどうか判定すれば良い.
二部グラフの判定手法としてUnionFindで行うのが良いというのをどこかで聞いたので以下を参考にした.
noshi91.hatenablog.com

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(int i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(int i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<int,int>;
using PP=pair<int,P>;
using Pl=pair<ll,ll>;
using PPl=pair<ll,Pl>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;
struct UnionFind{
    vector<int> data;
    UnionFind(int size):data(size,-1){}
    bool unite(int x,int y){
        x=root(x);y=root(y);
        if(x!=y){
            if(data[y]<data[x])swap(x,y);
            data[x]+=data[y];data[y]=x;
        }
        return x!=y;
    }
    bool find(int x,int y){
        return root(x)==root(y);
    }
    int root(int x){
        return data[x]<0?x:data[x]=root(data[x]);
    }
    int size(int x){
        return -data[root(x)];
    }
};
class Family{
public:
    string isFamily(vi a,vi b){
        int n=a.size();
        UnionFind uf(n*2);
        REP(i,n){
            if(a[i]!=-1){
                uf.unite(a[i],b[i]+n);
                uf.unite(a[i]+n,b[i]);
            }
        }
        REP(i,n){
            if(uf.find(i,i+n)){
                return "Impossible";
            }
        }
        return "Possible";
    }
};