蓝桥杯【day4】熄灯问题java版

题目描述：

解题代码：

package 第四天;

import java.util.Scanner;

public class 熄灯问题 {
    static byte[] orlight=new byte[5];//原来灯的情况
    static byte[] light=new byte[5];//变化中灯的情况
    static byte[] result=new byte[5];//结果
    static int n;

    static byte set(byte b,int i,int c){//指定b的i位为c
        if(c==0){
            b &=~(1<<i);
        }else {
            b|=(1<<i);
        }
        return b;
    }
    static int get(byte b,int i){//获取b的i位
        return (b>>i)&1;
    }
    static byte file(byte b,int i){//让b的i位取反
        b^=(1<<i);
        return b;
    }
    static void output(int i,byte [] result){
        System.out.printf("第%d场方案：\n",i);
        for(int p=0;p<5;p++){
            for(int q=0;q<6;q++){
                System.out.print(get(result[p],q)+" ");
            }
            System.out.println("\n");
        }
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        n = scanner.nextInt();
        for (int a = 0; a < n; a++) {
            for (int j = 0; j < 5; j++) {
                for (int p = 0; p < 6; p++) {
                    orlight[j] = set(orlight[j], p, scanner.nextInt());
                }
            }

            for (byte num = 0; num < 64; num++) {
                byte swich = num;
                light = orlight.clone();
                for (int i = 0; i < 5; i++) {
                    result[i] = swich;//第i行的开关的状态
                    for (int j = 0; j < 6; j++) {
                        if (get(swich, j) == 1) {
                            if (j > 0) {
                                light[i] = file(light[i], j - 1);
                            }
                            light[i] = file(light[i], j);
                            if (j < 5) {
                                light[i] = file(light[i], j + 1);
                            }
                        }
                    }
                    if (i < 4) {
                        light[i + 1] ^= swich;
                    }
                    swich = light[i];

                }
                if (light[4] == 0) {
                    output(n, result);
                    break;
                }
            }
        }
    }
}

 注意事项：

1.位运算：

 

 菜鸟教程链接：https://www.runoob.com/w3cnote/bit-operation.html​​​​​​

 算法思想：

这里用byte数来存放灯泡状态和开关个数。方便进行位运算。

因为第一行的开关状态可以决定第一行的开关状态，而第一行的灯泡可以决定第二行的开关状态

所以只用枚举第一行的开关状态就行，有2^6种可能性，联系到这里使用byte数来表示开关状态的，所以枚举只用设置成byte n=0~63(000000~111111)种可能。

最后light[i]=swich，是用i的灯泡状态来表示i+1的开关状态。