1049 棋盘染色

#include<cstdio>
#include<cstring>
using namespace std;
#define N 110
const int dx[]={0,0,1,-1};
const int dy[]={1,-1,0,0};
int n,sx,sy,tot,total;
bool a[N][6],vis[N][6];
void check_dfs(int x,int y){//遍历全图，统计染色点数
    tot++;
    vis[x][y]=1;
    for(int i=0;i<4;i++){
        int nx=x+dx[i];
        int ny=y+dy[i];//有的点只有被染色成连通块才会被统计到 
        if(!vis[nx][ny]&&nx>=1&&nx<=n&&ny>=1&&ny<=5&&a[nx][ny]) 
            check_dfs(nx,ny);
    }
}
bool can(int x){
    tot=0;//经过check_dfs后全图总共有tot个点被染色 
    memset(vis,0,sizeof vis);
    check_dfs(sx,sy);
    if(tot==total+x) return 1;//total为初始时染色点数，x为power_dfs之后多染的点数 
    return 0;
}
bool power_dfs(int x,int y,int now,int sum){//前两维为坐标，now表示当前比初始状态多涂了now个点，sum表示本次搜素的目标是涂sum个 
    if(now==sum){
        if(can(sum)) return 1;//判断是否已联通 
        return 0;
    }
    for(int j=y+1;j<=5;j++){//扫当前点的该行的右边 
        if(!a[x][j]){
            a[x][j]=1;
            if(power_dfs(x,j,now+1,sum)) return 1;
            a[x][j]=0;
        }
    }
    for(int i=x+1;i<=n;i++){//扫当前点的下方、左下方、右下方的所有区域 
        for(int j=1;j<=5;j++){
            if(!a[i][j]){
                a[i][j]=1;
                if(power_dfs(i,j,now+1,sum)) return 1;
                a[i][j]=0;
            }
        }
    }
    return 0;
}
int main(){
    //scanf("%d",&n);//先做的棋盘染色2，懒得改了，见谅 
    n=5;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=5;j++){
            scanf("%1d",&a[i][j]);
            if(a[i][j]){
                total++;
                sx=i;sy=j;
            }
        } 
    }
    for(int i=0;i<=5*n;i++){
        if(power_dfs(1,1,0,i)){
            printf("%d\n",i);
            return 0;
        }
    }
    return 0;
}