七段码问题(dfs以及并查集的应用)

题目

上图给出了七段码数码管的一个图示，数码管中一共有 77 段可以发光的二 极管，分别标记为 a, b, c, d, e, f, ga,b,c,d,e,f,g。
小蓝要选择一部分二极管（至少要有一个）发光来表达字符。在设计字符 的表达时，要求所有发光的二极管是连成一片的。
例如：bb 发光，其他二极管不发光可以用来表达一种字符。
例如 cc 发光，其他二极管不发光可以用来表达一种字符。这种方案与上 一行的方案可以用来表示不同的字符，尽管看上去比较相似。
例如：a, b, c, d, ea,b,c,d,e 发光，f, gf,g 不发光可以用来表达一种字符。
例如：b, fb,f 发光，其他二极管不发光则不能用来表达一种字符，因为发光 的二极管没有连成一片。
请问，小蓝可以用七段码数码管表达多少种不同的字符？

题解代码

#include<bits/stdc++.h>
using namespace std;
const int N=10;
int use[N],ans,e[N][N],fa[N];
void init(){
	/*	
		连边建图，e[i][j]==1表示i和j相邻
		a b c d e f g
		1 2 3 4 5 6 7
	*/
	e[1][2]=e[1][6]=1;
	e[2][1]=e[2][7]=e[2][3]=1;
	e[3][2]=e[3][4]=e[3][7]=1;
	e[4][3]=e[4][5]=1;
	e[5][4]=e[5][6]=e[5][7]=1;
	e[6][1]=e[6][5]=e[6][7]=1;
}
int find(int u){
	if(fa[u]==u)return u;
	fa[u]=find(fa[u]);
	return fa[u];
}//并查集
void dfs(int d){
	if(d>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(e[i][j]&&use[i]&&use[j]){
					int fx=find(i),fy=find(j);//fx！=fy就是两个不同的灯都亮的情况 
					if(fx!=fy)fa[fx]=fy;//如果不在同一集合,合并:此步将两个相邻的灯合并到同一个集合中 
				}
		int k=0;
		for(int i=1;i<=7;i++)
			if(use[i]&&fa[i]==i)k++;//如果一个灯都不亮则k=0，如果有多个灯亮而且有多组灯相连则k>1 
		if(k==1)ans++;//如果所有亮灯都属于同一个集合
		return;
	}
	use[d]=1;//打开d这个灯，继续开关下一个灯
	dfs(d+1);
	use[d]=0;//关闭d这个灯，继续开关下一个灯
	dfs(d+1);
}
int main(){
	init();
	dfs(1);
	cout<<ans;
}

解析

此题先是用深度优先搜索枚举了所有可能出现的亮灯情况
然后使用并查集来检测所有亮了的灯是否全部相连,其中并查集的合并操作确保了相邻的灯所属的集合相同,从而加下来的判断

for(int i=1;i<=7;i++)
			if(use[i]&&fa[i]==i)k++;//如果一个灯都不亮则k=0，如果有多个灯亮而且有多组灯相连则k>1 

只有当深度优先的目前情况下,只有一组相邻亮灯的情况下结果才加1.
巧妙地达成了所有亮灯都相连的条件!
妙啊!!!