POJ3279 翻硬币（状态压缩DFS）

每日打卡（1/1）

传送门：点击打开链接

题目大意：一个n*m的矩阵翻硬币，翻一枚硬币会涉及周围四个，求将所有硬币正面朝上最少需要翻哪几个硬币。

思路：

    与之前一题求最少要翻多少次不一样，这次要记录状态。

    首先我们先想到枚举第一排的翻法，然后对其他排时，只要上面一个硬币是反面朝上就翻，最后check一下。

第一个比较值得学习的地方就是如何枚举第一排的翻法：

    

for(int i=0;i<(1<<m);i++)
	 dfs(i);
for(int j=0;j<m;j++)
	{
		if((i>>j)&1) {
			flip(0,j);
			cnt++;
		}
	}

妙不可言啊，一下子就枚举出了一个2^m的全排列。

至于状态记录，可以每次都将数组a复制到数组temp，再新开一个数组vis，如果翻了某一枚，vis[x][y]就变成1，在最后检查时，如果满足情况，就把vis值复制到另一个数组res，因为vis数组是每次都要清0的，这样我们再遇到两组解比较大小时可以用res数组记录最优答案的解。

最后输出的时候，如果ans没有变化，说明没有解，输出impossible；

否则输出res数组。

底下附代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
using namespace std;
const int maxn = 18;
int m,n,ans=1e6,dir[4][2]={1,0,-1,0,0,1,0,-1};
bool a[maxn][maxn],vis[maxn][maxn],temp[maxn][maxn],res[maxn][maxn];

void flip(int x,int y)
{
	vis[x][y] = 1;
	temp[x][y] = !temp[x][y];
	if(x-1>=0) temp[x-1][y] = !temp[x-1][y];
	if(x+1<n) temp[x+1][y] = !temp[x+1][y];
	if(y-1>=0) temp[x][y-1] = !temp[x][y-1];
	if(y+1<m) temp[x][y+1] = !temp[x][y+1];
}

bool check()
{
	for(int i=0;i<n;i++)
	 for(int j=0;j<m;j++)
	  if(temp[i][j])
	   return false;
	return true;
}

void dfs(int k)
{
	memcpy(temp,a,sizeof(a));
	memset(vis,0,sizeof(vis));
	int cnt = 0;
	for(int j=0;j<m;j++)
	{
		if((k>>j)&1) {
			flip(0,j);
			cnt++;
		}
	}
	for(int i=1;i<n;i++)
	 for(int j=0;j<m;j++)
	  if(temp[i-1][j]==1) {
	  	flip(i,j);
	  	cnt++;
	  }
	if(check()&&cnt<ans) {
		ans = cnt;
		memcpy(res,vis,sizeof(vis));
	}
}

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	 for(int j=0;j<m;j++)
	  scanf("%d", &a[i][j]);
	for(int i=0;i<(1<<m);i++)
	 dfs(i);
	if(ans==1e6) cout<<"IMPOSSIBLE"<<endl;
	else for(int i=0;i<n;i++)
	{
		for(int j=0;j<m-1;j++)
		{
			cout<<res[i][j]<<" ";
		}
		cout<<res[i][m-1];
		cout<<endl;
	}
	return 0;
}