并查集+dfs——七段码(蓝桥杯真题)

《七段码》题目入口

 * 整体思路：
 * 1、所有灯管的情况列出来（要么亮，要么不亮）
 * 2、把亮的情况都用并查集连接起来（连的时候注意刚开始灯管的初始化）
 * 3、最后把连接起来的灯管进行判断
 *             ————如果只有一条线，则代表这个组合可以全亮
 *             ————如果有多条线，就说明组合链接不上，中间是断开的

import java.util.Arrays;
/*
 * 整体思路：
 * 1、所有灯管的情况列出来（要么亮，要么不亮）
 * 2、把亮的情况都用并查集连接起来（连的时候注意刚开始灯管的初始化）
 * 3、最后把连接起来的灯管进行判断
 * 			————如果只有一条线，则代表这个组合可以全亮
 * 			————如果有多条线，就说明组合链接不上，中间是断开的
 */
import java.util.Scanner;
public class Main {
	static Scanner sc = new Scanner(System.in);
	static int[][] a=new int[8][8];
	static int[] fa=new int[8];
	static boolean[] st=new boolean[8];
	static int ans;
	public static int find(int x) {
		if(fa[x]==x) return x;//如果自己就是本身了，那我就是祖宗
		return find(fa[x]);//找爸爸的爸爸
	}
	public static void dfs(int u) {
		if(u>7) {//七个灯管如果选择好了后
			for(int i=1;i<=7;i++)
	            fa[i]=i;//祖先都是自己，自成一派
			//并查集 把所有可以连接的连接起来
			for(int i=1;i<=7;i++) {
				for(int j=1;j<=7;j++) {
					if(a[i][j]==1&&st[i]&&st[j]) {//if成立就可以连接起来
						int xx=find(i);
						int yy=find(j);
						if(xx!=yy) {//不考虑遍历顺序带来的重复
							//因为1,2连接起来后，1,2的祖宗都是2了
							//下一次遍历顺序是2,1，他俩的祖宗是一样的，所以不用连了
							fa[xx]=yy;							
						}
					}
				}
			}
			int k=0;
			for(int i=1;i<=7;i++) {
				if(fa[i]==i&&st[i]) {
					k++;
				}
			}
			if(k==1) ans++;
			return;
		}
		st[u]=false;
		dfs(u+1);
		st[u]=true;
		dfs(u+1);
	}
	public static void main(String[] args) throws Exception{
		a[1][2]=a[1][6]=1;
	    a[2][1]=a[2][7]=a[2][3]=1;
	    a[3][7]=a[3][4]=a[3][2]=1;
	    a[4][3]=a[4][5]=1;
	    a[5][7]=a[5][4]=a[6][5]=1;
	    a[6][1]=a[6][7]=a[5][6]=1;
	    a[7][2]=a[7][3]=a[7][5]=a[7][6]=1;
	    dfs(1);
	    System.out.println(ans);
	}
}