OpenJudge-NOI/2.1基本算法之枚举-1815:画家问题
1815:画家问题
总时间限制: 1000ms 内存限制: 65536kB
描述
有一个正方形的墙，由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题解
此题和熄灯问题十分类似，可以说是相当于熄灯问题的一个翻版，还是采用局部思想的方法。此题的局部还是第一行，我们只需枚举第一行所有砖块的涂画情况，以此来推断出下一行的的涂画情况（使每行的砖块都被涂画成黄色）直到最后一行，然后检查一下最后一行的所有砖块是否能被涂成黄色。
若要观看详细的题解，还请读者移步到我写的上一篇文章熄灯问题-枚举。
这里再说一说这道题与熄灯问题的不同之处。这道题的具体实现要比熄灯问题多了一个变量，这个变量（count）用来统计需要涂画的砖块数量，并且题目中说是最少需要涂画的砖数，所以还要进行比较取最小值，因此还要有一个变量来记录最小值（mincount）。
通过上述的简单分析，以下是具体的实现方案和代码。
方案一 采用二维数组
#include<iostream>
#include<cstring>
using namespace std;
int oribricks[16][17];
int draw[16][17];
int n;
int guess()
{
int r,c;
for(r=1; r<n; r++)
for(c=1; c<n+1; c++)
{
draw[r+1][c]=(draw[r][c]+draw[r][c-1]+draw[r][c+1]+draw[r-1][c]+oribricks[r][c])%2;
}
for(c=1; c<n+1; c++)
{
if(oribricks[n][c]!=(draw[n][c]+draw[n][c-1]+draw[n][c+1]+draw[n-1][c])%2)
return 250;
}
int count=0;
for(r=1; r<n+1; r++)
{
for(c=1; c<n+1; c++)
{
if(draw[r][c]==1)
count++;
}
}
return count;
}
int enumerate()
{
int c;
int count=0,mincount=250;
while(draw[1][0]<1)
{
count=guess();
if(count<mincount)
mincount=count;
draw[1][n]++;
c=n;
while(draw[1][c]>1)
{
draw[1][c]=0;
c--;
draw[1][c]++;
}
}
return mincount;
}
int main()
{
int r,c,mincount=0;
cin >> n;
memset(oribricks,0,sizeof(oribricks));
memset(draw,0,sizeof(draw));
for(r=1; r<n+1; r++)
{
for(c=1; c<n+1; c++)
{
char color;
cin >> color;
oribricks[r][c]=(color=='w')? 1 : 0;
}
}
mincount=enumerate();
if(mincount==250)
cout << "inf";
else
cout << mincount;
return 0;
} 代码注释：
/*
w:white 白 白色的砖用 1表示
y:yellow 黄 黄色的砖用 0表示
*/
#include<iostream>
#include<cstring>
using namespace std;
int oribricks[16][17];//读入的墙的初始状态 1表示白色砖，0表示黄色砖
int draw[16][17];//涂画的结果，1表示涂，0表示不涂
int n;//墙的大小 (1≤n≤15)
int guess()//判断涂画好第一行砖块的颜色后，推算出最后一行的砖块是否都能被涂画成黄色
{
int r,c;
for(r=1; r<n; r++)
for(c=1; c<n+1; c++)
{
draw[r+1][c]=(draw[r][c]+draw[r][c-1]+draw[r][c+1]+draw[r-1][c]+oribricks[r][c])%2;
}
for(c=1; c<n+1; c++)
{
if(oribricks[n][c]!=(draw[n][c]+draw[n][c-1]+draw[n][c+1]+draw[n-1][c])%2)
return 250;//最后一行的砖块若不能都被涂成黄色则返回250
}
int count=0;
for(r=1; r<n+1; r++)
{
for(c=1; c<n+1; c++)
{
if(draw[r][c]==1)
count++;
}
}
return count;//若最后一行的所有砖都能被涂成黄色，则函数返回需要涂画的砖块数，也就是draw数组中 1的数量（1代表涂）
}
int enumerate()//枚举第一行进行涂画的状态数，将draw数组第一行看成一个二进制数，左边高位，右边低位
{
int c;
int count=0,mincount=250;
while(draw[1][0]<1)
{
count=guess();
if(count<mincount)
mincount=count;//通过比较，取最少需要涂画的砖数
draw[1][n]++;
c=n;
while(draw[1][c]>1)
{
draw[1][c]=0;
c--;
draw[1][c]++;
}
}
return mincount;
}
int main()
{
int r,c,mincount=0;
cin >> n;
memset(oribricks,0,sizeof(oribricks));
memset(draw,0,sizeof(draw));
for(r=1; r<n+1; r++)//读入墙的每块砖初始的颜色并将它转换为数字 1或 0
{
for(c=1; c<n+1; c++)
{
char color;
cin >> color;
oribricks[r][c]=(color=='w')? 1 : 0;//1表示白色砖，0表示黄色砖
}
}
mincount=enumerate();
if(mincount==250)
cout << "inf";
else
cout << mincount;
return 0;
} 这里再解释一下为什么guess()函数中最后一行的砖块若不能都被涂成黄色则返回250，其实这很简单，因为墙的大小最大为15，15*15=225，涂画的最大砖块数为225（其实比225更小），所以返回一个不可能涂画到的数目以便比较取最小值。若到最后枚举循环完成后返回的最小值还为250，那么就找不到任何一种涂画的方案使得墙的所有砖块被涂画成黄色，那么则输出“ inf ”。
还有为什么在enumerate()函数中循环结束条件表达式为draw[1][0]<1，因为和熄灯问题类似，为了简化计算数组下一行值的计算公式，我们将最大行数多加1，列数多加2，相当于把这个墙除了最后一行围了一圈。而draw[1][0]就是一个边界条件，我们把draw数组第一行模拟成一个二进制数，最高位其实就是draw[1][1]，模拟二进制加法进位，比如当二进制数为01111时，那么再加1时就成了10000。其中10000中的1相当于就是draw[1][0]=1，故当draw[1][0]=1时第一行枚举完成，所以跳出while循环。
方案二 采用位运算
位运算详细说明还请看我写的上一篇文章熄灯问题，故这里不会解释太多。
代码如下：
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n;
int GetBit(short c,int i)
{
return (c >> i) & 1;
}
void SetBit(short & c,int i,int v)
{
if(v)
c |= (1 << i);
else
c &= ~(1 << i);
}
void Flip(short & c,int i)
{
c ^= (1 << i);
}
int onecount(short x)
{
int c;
for(c=0; x>0; c++)
{
x &= (x-1);
}
return c;
}
int main()
{
cin >> n;
short oribricks[16];
short draw[16];
memset(oribricks,0,sizeof(oribricks));
for(int r=0; r<n; r++)
{
for(int c=0; c<n; c++)
{
char color;
cin >> color;
SetBit(oribricks[r],c,color!='y');
}
}
int num = 1 << n;
int mincount = 1 << 30;//2^30
int a;
short bricks;
for(a=0; a<num; a++)
{
memcpy(draw,oribricks,sizeof(oribricks));
bricks=a;
int count=0;
for(int r=0; r<n; r++)
{
count += onecount(bricks);
for(int c=0; c<n; c++)
{
if(GetBit(bricks,c))
{
if(c>0)
Flip(draw[r],c-1);
Flip(draw[r],c);
if(c<n-1)
Flip(draw[r],c+1);
}
}
if(r<n-1)
draw[r+1] ^= bricks;
bricks = draw[r];
}
if(draw[n-1]==0)
{
mincount=min(count,mincount);
}
}
if(mincount == 1 << 30)
cout << "inf";
else
cout << mincount;
return 0;
}代码解释：
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n;//墙的大小 (1≤n≤15)
int GetBit(short c,int i)//取c的第i位
{
return (c >> i) & 1;
}
void SetBit(short & c,int i,int v)//设置c的第i位为v
{
if(v)
c |= (1 << i);
else
c &= ~(1 << i);
}
void Flip(short & c,int i)//将c的第i位取反
{
c ^= (1 << i);
}
int onecount(short x)//统计x中 1的个数，即需要涂画的砖块数
{
int c;
for(c=0; x>0; c++)
{
x &= (x-1);
}
return c;
}
int main()
{
cin >> n;
short oribricks[16];//存储最初读入的每块砖的颜色，一个比特代表一块砖
short draw[16];//不停变化的墙矩阵
memset(oribricks,0,sizeof(oribricks));
for(int r=0; r<n; r++)//读入初始每块砖的颜色
{
for(int c=0; c<n; c++)
{
char color;
cin >> color;
SetBit(oribricks[r],c,color!='y');//1表示白色，0表示黄色 w:1 y:0
}
}
int num = 1 << n;//左移1位相当于乘以2，1 << n 表示首行砖块能够进行涂画的状态数
int mincount = 1 << 30;//2^30（2的30次方）
int a;
short bricks;//用来存储每行砖块该怎么涂，从二进制看1表示涂画，0表示不涂画
for(a=0; a<num; a++)//遍历首行砖块进行涂画的所有状态
{
memcpy(draw,oribricks,sizeof(oribricks));
bricks=a;//首行砖块涂画状态
int count=0;//统计需要涂画的砖块数量
for(int r=0; r<n; r++)
{
count += onecount(bricks);//累加每行需要涂画的砖块数量
for(int c=0; c<n; c++)
{
if(GetBit(bricks,c))
{
if(c>0)
Flip(draw[r],c-1);//改左边砖块颜色
Flip(draw[r],c);//改本位置砖块颜色
if(c<n-1)
Flip(draw[r],c+1);//改右边砖块颜色
}
}
if(r<n-1)
draw[r+1] ^= bricks;//改下一行砖块颜色
bricks = draw[r];//下一行该怎么涂画已被上一行涂画完的状态确定（也就是第i+1行每块砖涂画的方案和第i行每块砖颜色相同）
}
if(draw[n-1]==0)
{
mincount=min(count,mincount);//通过比较取最小值，也就是题目中说的最少需要涂画的砖数
}
}
if(mincount == 1 << 30)
cout << "inf";
else
cout << mincount;
return 0;
}最后再说一下，onecount()也就是统计二进制数中的1的个数这个函数有多种实现方法，大家可以自己去写一下或者搜索其他博主的写法。
位运算牛逼！！！
此题来源于OpenJudge，链接如下：
http://noi.openjudge.cn/ch0201/1815/
感谢观看！若有错误还请指出，如果有什么不懂的也可以留言。
234
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
