OPJ2.1基本算法之枚举
1815:画家问题
题目传送门
描述
有一个正方形的墙,由N*N个正方形的砖组成,其中一些砖是白色的,另外一些砖是黄色的。Bob是个画家,想把全部的砖都涂成黄色。但他的画笔不好使。当他用画笔涂画第(i, j)个位置的砖时, 位置(i-1, j)、 (i+1, j)、 (i, j-1)、 (i, j+1)上的砖都会改变颜色。请你帮助Bob计算出最少需要涂画多少块砖,才能使所有砖的颜色都变成黄色。
输入
第一行是一个整数n (1≤n ≤15),表示墙的大小。接下来的n行表示墙的初始状态。每一行包含n个字符。第i行的第j个字符表示位于位置(i,j)上的砖的颜色。“w”表示白砖,“y”表示黄砖。
输出
一行,如果Bob能够将所有的砖都涂成黄色,则输出最少需要涂画的砖数,否则输出“inf”。
样例输入
5
wwwww
wwwww
wwwww
wwwww
wwwww
样例输出
15
**思路:**和之前的熄灯问题一样,这道题的思路也是枚举第一行的状态,从而推得其他块的状态。
AC代码:
#include <bits/stdc++.h>
using namespace std;
int a[20][20],b[2][20],c[20][20],n,tot,minn,sign,d[5][5]={{0,0},{1,0},{-1,0},{0,1},{0,-1}};
bool check()
{
for(int j=1;j<=n;j++)
if(a[n][j]==0)return false;
return true;
}
void op(int x,int y)
{
int row,col;
for(int i=0;i<5;i++)
{
row=x+d[i][0],col=y+d[i][1];
if(row>=1&&row<=n&&col>=1&&col<=n)
a[row][col]=!a[row][col];
}
}
void solve()
{
for(int i=1;i<=n;i++)
if(b[1][i]==1){op(1,i);tot++;}
for(int i=2;i<=n;i++)
for(int j=1;j<=n;j++)
if(a[i-1][j]==0)
{
op(i,j);
tot++;
}
if(check())
{
sign=1;
if(tot<minn)minn=tot;
}
else
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
a[i][j]=c[i][j];
tot=0;
}
}
int main()
{
char ch;int t;
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
cin>>ch;
a[i][j]=(ch=='w')?0:1;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
c[i][j]=a[i][j];
minn=300,sign=0,tot=0;
for(int i=0;i<(int)pow(2,n);i++)
{
solve();
//枚举第一行的所有情况,枚举方法:用二进制依次加1的进位方法模拟实现
b[1][1]++;
t=1;
while(b[1][t]>1)
{
b[1][t]=0;
t++;
b[1][t]++;
}
}
if(sign)printf("%d",minn);
else printf("inf");
return 0;
}
代码我认为唯一难理解的地方已标出注释,如有问题可参考我熄灯问题的博客。