原题的链接95. 费解的开关 - AcWing题库
题目的大体意思即是把一副5*5方形中的所有0变为1,每次可以更改一个点的值0变成1,1变成0,相应的将上下左右的都进行更改。
遇到类似的题目都要考虑去使用位运算,因为位运算时间复杂度可以控制为1<<20,5*5的图就没法进行暴力,可以采取dfs的方式。
用位运算枚举出来第一排会出现的情况,用二进制码来表示,倘若是1就按开关,否则就不动,然后每一位的底下那个可以改变上面的状态,进行dfs即可。
代码如下
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n,ans;
char a[20][20],b[20][20];
void change(int x,int y){
b[x][y]^=1;
b[x-1][y]^=1;
b[x+1][y]^=1;
b[x][y-1]^=1;
b[x][y+1]^=1;
}
void dfs(){
ans=1e9;
for(int k=0;k<1<<5;k++){
memcpy(b,a,sizeof a);
int step=0;
for(int j=0;j<5;j++){
if(k>>j&1){
step++;
change(0,j);
}
}
int flag=1;
for(int i=0;i<4;i++){
for(int j=0;j<5;j++){
if(b[i][j]=='0'){
step++;
change(i+1,j);
}
}
}
for(int i=0;i<5;i++){
if(b[4][i]=='0'){
flag=0;
break;
}
}
if(flag==1) ans=min(step,ans);
}
}
int main(){
cin>>n;
while(n--){
for(int i=0;i<5;i++){
cin>>a[i];
}
dfs();
if(ans>6) cout<<-1<<endl;
else cout<<ans<<endl;
}
}