题目链接:https://www.acwing.com/problem/content/97/
主要思想:枚举第一行的状态,每一行都可以确定下一行的状态,最后判断是否最后一行灯被全部点亮即可。
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<string.h>
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<<doit()<<endl;
}
}