[POJ1681]Painter's Problem（高斯消元+dfs）

题目描述

传送门
题意：同POJ1222，如果有多解，输出修改点数最少的，如果无解，输出inf

题解

和POJ1222很类似，只需要再判断多解和无解的情况
如果出现自由元的话，就看它后面消出来的如果和b相等，那么就是有多解，如果和b不相等就是无解
因为要输出点数最少的，所以需要将所有的自由元的取值指数级暴搜一下
bitset传参不知道为什么老是炸掉，所以干脆回溯了…

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<bitset>
using namespace std;
#define N 300

char s[N][N];
int T,n,Min;
bitset<N>a[N];int b[N],ans[N];
bool flag;

void clear()
{
    n=0;Min=N;flag=false;
    for (int i=1;i<N;++i) a[i].reset();
    memset(b,0,sizeof(b));memset(ans,0,sizeof(ans));
}
void check()
{
    int now=0;
    for (int i=1;i<=n;++i) now+=ans[i];
    Min=min(Min,now);
}
void dfs(int dep)
{
    if (!dep)
    {
        check();
        return;
    }
    if (flag) return;
    if (a[dep][dep])
    {
        ans[dep]=b[dep];
        if (ans[dep])
        {
            for (int i=dep-1;i>=1;--i)
                if (a[i][dep]) b[i]^=ans[dep];
        }
        dfs(dep-1);
        if (ans[dep])
        {
            for (int i=dep-1;i>=1;--i)
                if (a[i][dep]) b[i]^=ans[dep];
        }
    }
    else
    {
        if (b[dep]) {flag=1;return;}
        ans[dep]=0;
        dfs(dep-1);
        ans[dep]=1;
        for (int i=dep-1;i>=1;--i)
            if (a[i][dep]) b[i]^=ans[dep];
        dfs(dep-1);
        for (int i=dep-1;i>=1;--i)
            if (a[i][dep]) b[i]^=ans[dep];
    }
}
void gauss()
{
    for (int i=1;i<=n;++i)
    {
        if (!a[i][i])
        {
            int num=i;
            for (int j=i+1;j<=n;++j)
                if (a[j][i]) {num=j;break;}
            if (num==i) continue;
            swap(a[i],a[num]);swap(b[i],b[num]);
        }
        for (int j=i+1;j<=n;++j)
            if (a[j][i]) a[j]^=a[i],b[j]^=b[i];
    }
    dfs(n);
}
int main()
{
    scanf("%d",&T);
    while (T--)
    {
        clear();
        scanf("%d",&n);
        for (int i=1;i<=n;++i) scanf("%s",s[i]+1);
        for (int i=1;i<=n;++i)
            for (int j=1;j<=n;++j)
            {
                int now=(i-1)*n+j;
                a[now][now]=1;
                if (i!=1) a[now][now-n]=1;
                if (i!=n) a[now][now+n]=1;
                if (j!=1) a[now][now-1]=1;
                if (j!=n) a[now][now+1]=1;
                if (s[i][j]!='y') b[now]=1;
            }
        n*=n;
        gauss();
        if (flag) puts("inf");
        else printf("%d\n",Min);
    }
}