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/
感谢观看!若有错误还请指出,如果有什么不懂的也可以留言。