BZOJ 2303 方格染色(带权并查集)

要使得每个2*2的矩形有奇数个红色，如果我们把红色记为1，蓝色记为0，那么我们得到了这2*2的矩形里的数字异或和为1.

对于每个方格则有a(i,j)^a(i-1,j)^a(i,j-1)^a(i-1,j-1)=1.由这些方程可以推出对于每个方格:

如果i,j都是偶数，则有a(i,j)^a(1,1)^a(i,1)^a(1,j)=1.

否则，a(i,j)^a(1,1)^a(i,1)^a(1,j)=0.枚举a(1,1)的染色情况。可以由a(i,j)的染色情况推出a(i,1)和a(1,j)是否颜色相同或者相反。

类似于a bug's life。那么把它们用带权并查集维护后，染色的种数就是一些联通块的染色种数，因为这些联通块互相不影响。

那么总染色数就是2^(cnt-1).

# include <cstdio>
# include <cstring>
# include <cstdlib>
# include <iostream>
# include <vector>
# include <queue>
# include <stack>
# include <map>
# include <set>
# include <cmath>
# include <algorithm>
using namespace std;
# define lowbit(x) ((x)&(-x))
# define pi acos(-1.0)
# define eps 1e-
# define MOD
# define INF
# define mem(a,b) memset(a,b,sizeof(a))
# define FOR(i,a,n) for(int i=a; i<=n; ++i)
# define FO(i,a,n) for(int i=a; i<n; ++i)
# define bug puts("H");
# define lch p<<,l,mid
# define rch p<<|,mid+,r
# define mp make_pair
# define pb push_back
typedef pair<int,int> PII;
typedef vector<int> VI;
# pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long LL;
int Scan() {
    int x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
void Out(int a) {
    if(a<) {putchar('-'); a=-a;}
    if(a>=) Out(a/);
    putchar(a%+'');
}
const int N=;
//Code begin...

struct Node{int l, r, c;}node[N];
int fa[N], dis[N];

LL pow_mod(LL a, LL n, LL mod){
    LL ret=, temp=a%mod;
    while (n) {
        if (n&) ret=ret*temp%mod;
        temp=temp*temp%mod;
        n>>=;
    }
    return ret;
}
int find(int x){
    int tmp;
    if (fa[x]!=x) tmp=find(fa[x]), dis[x]=dis[x]^dis[fa[x]], fa[x]=tmp;
    return fa[x];
}
bool union_set(int x, int y, int d){
    int u=find(x), v=find(y);
    if (u!=v) {dis[u]=dis[y]^d^dis[x]; fa[u]=v; return true;}
    else return dis[x]^dis[y]^d==;
}
int main ()
{
    int n, m, k, u, v, flag;
    LL ans=;
    scanf("%d%d%d",&n,&m,&k);
    FOR(i,,k) scanf("%d%d%d",&node[i].l,&node[i].r,&node[i].c);
    mem(dis,); flag=true; FOR(i,,n+m-) fa[i]=i;
    FOR(i,,k) {
        if (node[i].l==) u=;
        else u=m+node[i].l-;
        v=node[i].r;
        if (node[i].l%==&&node[i].r%==) {
            if (!union_set(u,v,node[i].c^)) {flag=false; break;}
        }
        else {
            if (!union_set(u,v,node[i].c)) {flag=false; break;}
        }
    }
    if (flag) {
        int cnt=;
        FOR(i,,n+m-) if (fa[i]==i) ++cnt;
        ans=(ans+pow_mod(,cnt-,MOD))%MOD;
    }
    mem(dis,); flag=true; FOR(i,,n+m-) fa[i]=i;
    FOR(i,,k) {
        if (node[i].l==) u=;
        else u=m+node[i].l-;
        v=node[i].r;
        if (node[i].l%==&&node[i].r%==) {
            if (!union_set(u,v,node[i].c)) {flag=false; break;}
        }
        else {
            if (!union_set(u,v,node[i].c^)) {flag=false; break;}
        }
    }
    if (flag) {
        int cnt=;
        FOR(i,,n+m-) if (fa[i]==i) ++cnt;
        ans=(ans+pow_mod(,cnt-,MOD))%MOD;
    }
    printf("%lld\n",ans);
    return ;
}