算法-熄灯问题

package MOOC;

import java.nio.channels.SelectableChannel;
import java.util.Scanner;

/**
题目描述
 * 有一个由按钮组成的矩阵，其中每行有6个按钮，共5行。每个按钮的位置上有一盏灯。
 * 当按下一个按钮后，该按钮以及周围位置(上边、下边、左边、右边)的灯都会改变一次。
 * 即，如果灯原来是点亮的，就会被熄灭； 如果灯原来是熄灭的，则会被点亮。
 * 在矩阵角上的按钮改变3盏灯的状态；在矩阵边上的按钮改变4盏灯的状态；其他的按钮改变5盏灯的状态。
 * 根据上面的规则，我们知道1）第2次按下同一个按钮时，将抵消第1次按下时所产生的结果。
 * 因此，每个按钮最多只需要按下一次；2）各个按钮被按下的顺序对最终的结果没有影响；3）对第1行中每盏点亮的灯，按下第2行对应的按钮，就可以熄灭第1行的全部灯。
 * 如此重复下去，可以熄灭第1、2、3、4行的全部灯。
 * 同样，按下第1、2、3、4、5列的按钮，可以熄灭前5列的灯。
输入样例
	0 1 1 0 1 0 
	1 0 0 1 1 1 
	0 0 1 0 0 1 
	1 0 0 1 0 1  
	0 1 1 1 0 0	
输出样例
	1 0 1 0 0 1 
	1 1 0 1 0 1 
	0 0 1 0 1 1 
	1 0 0 1 0 0 
	0 1 0 0 0 0
	输入样例中1代表亮着，0代表熄灭，输出样例中1代表按下，0代表不按，要求怎样按灯，可以使所有的灯都熄灭
 * @author Vivinia
 *
 * 2018年1月24日
 */
public class XiDeng {
	//枚举第一列，状态数2^6=64（每位两种状态，每行六位），第一列确定后后边几列都确定
	static char[] orLights=new char[5];    //每个元素对应灯矩阵的一行
	static char[] changing=new char[5];    //变化中的灯的矩阵
	static char[] result=new char[5];     //最终结果
	public static void main(String[] args) {
		Scanner input=new Scanner(System.in);
		for(int i=0;i<5;i++) 
			for(int j=0;j<6;j++) {
				int s=input.nextInt();        //读每个灯状态，s为0或1
				orLights[i]=setBit(orLights[i],j,s);      //将s放入初始数组中
			}
		input.close();
		for(int n=0;n<64;n++) {      //枚举0到63对应01组合，作为第一行的可能
			int switchs=n;            //当前行的开关状态
			System.arraycopy(orLights, 0, changing, 0, 5);    //orLights数组复制到changing，如果每一个结果不对则从头开始
			for(int i=0;i<5;i++) {
				result[i]=(char) switchs;      //结果中第i行的开关状态
				for(int j=0;j<6;j++) {          //对第i行的灯进行处理，j代表第i行的第j个开关
					if(getBit((char)switchs,j)==1) {        //如果第j个开关为1，代表要按
						if(j>0)        //如果j不是每行第一个和最后一个开关，则影响本身加左右的
							changing[i]=flipBit(changing[i],j-1);    //j左边开关
						changing[i]=flipBit(changing[i],j);     //j开关
						if(j<5)             //j不是每行最后一个开关，也影响右边开关
							changing[i]=flipBit(changing[i],j+1);     //j右边开关
					}
				}
				if(i<4)        //如果i不是最后一行，需要处理i+1行的灯的情况
					changing[i+1]^=switchs;     //下一行和上一行做抑或，不一样为1，一样为0，这是因为上一行的已经按了，为1是要按的，所以对应下边要取反（和1异或取反）
				switchs=changing[i];   //将当前行设为swiychs，用来对照着按下一行（里边的1必须全部被按掉）
			}
			if(changing[4]==0) {            //最后一行等于0，代表全部熄灭
				outPutResult(result);
				break;
			}
		}
	}
	
	//取字符c的第i个比特
	static int getBit(char c,int i) {
		return (c>>i)&1;           //将c右移i位，第i位移到最右边，和1相与，结果为0或者1
	}
	static //字符c的第i位设置为v
	char setBit(char c,int i,int v) {
		if(v==1)
			c|=(1<<i);       //c等于c和1<<i相或，i位就变成了1
		else 
			c&=~(1<<i);      //c等于c和1<<i取反相与，i位就变成了0
		return c;    
	}
	static //字符c第i位反转
	char flipBit(char c,int i) {
		c^=(1<<i);      //c和1<<i相异或，一样（都为1）为0，不一样（第i位为0）为1
		return c;
	}
	static //输出结果
	void outPutResult(char result[]) {
		for(int i=0;i<5;i++) {
			for(int j=0;j<6;j++) {
				System.out.print(getBit(result[i], j));    //输出result数组每个元素的第j个字符);
				if(j<5)
					System.out.print(" ");      //每个数之间有空格
			}
			System.out.println();     //一行数据就换行
		}
	}
}                                                                  

代码是仿照视频中的，加了比较详细的注释，很佩服这段代码，学到了很多，以后想灵活运用还得多练习：

Java中可以使用位运算（我渣渣，之前并不清楚，也从来没用过）；