费解的开关【二进制枚举】

题目链接:https://www.acwing.com/problem/content/97/

主要思想:枚举第一行的状态,每一行都可以确定下一行的状态,最后判断是否最后一行灯被全部点亮即可。

#include
#include
#include
#include
#include
using namespace std;

int map[10][10];
const int inf = 0x3f3f3f3f;
int dx[5]={0,0,1,-1,0};
int dy[5]={1,-1,0,0,0};
void action(int x,int y)
{
	for(int i=0;i<5;i++)
	{
		int nx = x+dx[i];
		int ny = y+dy[i];
		if(nx<0 || nx>=5 || ny<0 || ny>=5) continue;
		map[nx][ny]^=1;
	}
}
int doit() 
{
	int ans = inf;
	for(int i=0;i< 1<<5 ;i++)   //00000--->11111 枚举第一行所有状态
	{
		
		int res = 0;
		int back[10][10];
		memcpy(back,map,sizeof(map));
		for(int j=0;j<5;j++)
		{
			if(i>>j&1)
			{
				res++;
				action(0,j);
			}
		}
		
		for(int j=0;j<4;j++)
		{
			for(int k=0;k<5;k++)
			{
				if(map[j][k]==0)
				{
					res++;
					action(j+1,k);
				}
			}
		}
		bool ok = true;
		for(int j = 0;j<5;j++)
		{
			if(map[4][j]==0)
			{
				ok = false;
				break;
			}
		}
		if(ok) ans = min(ans,res);
		memcpy(map,back,sizeof(back));
	}
	if(ans>6) return -1;
	return ans;
	 
}
int main()
{
	int T;
	cin>>T;
	
	while(T--)
	{getchar();
		for(int i=0;i<5;i++)
		{
			for(int j=0;j<5;j++)
			{
				char ch = getchar();
				map[i][j]=ch-'0';
			}
			getchar();
		}
		
		cout<

 

你可能感兴趣的:(AcWing)