【蓝桥杯省赛】七段码，二进制枚举 + 判断连通性（DFS或并查集）

原题

小蓝要用七段码数码管来表示一种特殊的文字。上图给出了七段码数码管的一个图示，数码管中一共有 7 段可以发光的二 极管，分别标记为 a, b, c, d, e, f, g。 小蓝要选择一部分二极管（至少要有一个）发光来表达字符。在设计字符 的表达时，要求所有发光的二极管是连成一片的。

例如：b 发光，其他二极管不发光可以用来表达一种字符。

例如：c 发光，其他二极管不发光可以用来表达一种字符。这种方案与上一行的方案可以用来表示不同的字符，尽管看上去比较相似。

例如：a, b, c, d, e 发光，f, g 不发光可以用来表达一种字符。

例如：b, f 发光，其他二极管不发光则不能用来表达一种字符，因为发光 的二极管没有连成一片。

请问，小蓝可以用七段码数码管表达多少种不同的字符？

答案：80

分析

在此之前，我们先将七段码抽象为无向图。

问题就相当于：求上图的连通块个数！每个连通块所包含的节点个数可能是1~7个

一开始想着怎么搜索，但是感觉好难写。转换一下思维：在上图中，我们可以选出任意多个二极管来发光，其他不选的则不发光。然后判断选出的二极管是否构成一个连通块。如果是连通块，则答案+1。

1、选出任意多个二极管有多少种方案？

对于每一个二极管，我们要么选，要么不选，共有7个二极管，所以有 2^7-1 种方案。之所以减1，是因为要排除，都不选的这一种情况。将7个二极管编号0~6，我们可以用二进制来表示每一种情况。

比如：111 1111 表示所有二极管都选；

000 1111 表示编号为 0、1、2、3 的二极管选中发光，其他不选（不发光）。

那么所有的情况：000 001 ~ 111 1111。即：1~127

2、如何判断多个节点是否属于同一个连通块？

DFS 或者 并查集

DFS判断图是否连通

1. 从某个结点 v 出发，访问结点 v，并令 vis[v] = 1
2. 查找 v 的所有邻接点 i，若结点 i 并未被访问过（vis[i] = 0），则从结点 i 出发，继续深搜，转至步骤1
3. 深搜完整个图之后，遍历 vis 数组。如果数组全是1，说明只有一个连通分量，所有节点同属一个连通块。否则就存在多个连通分量

并查集判断图是否连通

默认你已经学习过并查集

1. 初始化每个结点的根节点为结点本身（开始时各个节点分别属于不同集合），即：parent[i]=i
2. 从某个结点 v 开始，查找 v 的邻接点 i，如果 v 的根节点和 i 的根节点不同（节点 v 和节点 i 属于不同集合），则将 i 的根节点指向 v 的根节点（或者将v 的根节点指向 i 的根节点）（合并两个集合）
3. 遍历完整个图后，如果所有节点的根节点都一样，则表示所有节点属于同一个集合（连通分量、连通块）。否则就存在多个连通分量

代码

代码中有注释，有些细节处的代码需要注意和思考！

代码中也有一些为了调试，添加的一些打印代码。

个人比较推荐：DFS。比较简单，不容易出错。

DFS

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;

vector<int> g[10];
int lighted[10];
int vis[10]; 

//0 1
//1 2
//2 3
//3 4
//4 5
//5 0
//1 6
//2 6
//4 6
//5 6

void dfs(int p) {
	vis[p] = 2; // 搜过 
	
	// 与p相邻的点 
	for(int i=0; i<g[p].size(); i++) {
		int near = g[p][i];
		if (vis[near]==1) { // 且又是选中点亮的二极管，且未搜过 
			dfs(near);
		}
	}
}

// 检查是否连通 
int isBlock(int kase) { // 某一种点亮的情况 
	
	
	memset(vis, 0, sizeof(vis));
	
	int k = 0; // 已选择点亮的二极管的数量 
	for (int i=0; i<7; i++) {
		int bit = (kase>>i) & 1;
		if (bit) { // 编号为i的二极管点亮 
			lighted[k++] = i;
			vis[i] = 1;
		} 
	}
	
//	for(int i=k-1; i>=0; i--) {
//		printf("%d ", lighted[i]);
//	}
	
	
	dfs(lighted[0]);
	
	// 未点亮0，点亮为搜过的为1，点亮且搜过的为2
	// 统计2的个数
	int cnt = 0;
	for (int i=0; i<7; i++) {
		if(vis[i]==2) {
			cnt++;
		}
	}
	
//	if(cnt == k) {
//		printf("是\n");
//	} else {
//		printf("否\n");
//	}
	
	return cnt == k; // 相等就是一个完整的连通块	
} 

int input() {
	int a, b;
	while (cin>>a>>b) {
		g[a].push_back(b);
		g[b].push_back(a);
	}
	
//	for(int i=0; i<7; i++) {
//		printf("%d = ", i);
//		for(int j=0; j<g[i].size(); j++) {
//			printf("%d ", g[i][j]);
//		}
//		printf("\n");
//	}
	
}

int main() {
	
	input();
	int ans = 0;
	int n = 1<<7;
	// 6 5 1 0
//	printf("%d\n", isBlock(99));
	
	for(int i=1; i<n; i++) {
		ans += isBlock(i);
	}
	
	printf("%d\n", ans);
	
	return 0;
}

并查集

#include<iostream>
#include<cstdio>
#include<vector>
#include<set>
using namespace std;

int g[10][7]; 
int par[10];
int lighted[10];

int find(int x) {
	return par[x]==x ? x : (par[x]=find(par[x])); 
}

void merge(int cur, int k) {
	for (int i=0; i<k; i++) {
		int t = lighted[i];
		int root1 = find(t);
		int root2 = find(cur);
		// 如果t和cur不在同一个连通块，且t所在连通块至少有一个点与cur相连，则将cur合并至连通块 
		if (root1!=root2 && g[t][cur] && g[cur][t])  {
			par[root1] = root2;
		}
	} 
}

//0 1
//1 2
//2 3
//3 4
//4 5
//5 0
//1 6
//2 6
//4 6
//5 6

// 检查是否连通 
int isBlock(int kase) { // 某一种点亮的情况 
	
	// 存储点亮的二级管的编号 
	// 并查集中的parent数组。每种情况都要重新初始化 
	for(int i=0; i<7; i++) {
		par[i] = i;
	}
	
	int k = 0; // 已选择点亮的二极管的数量 
	 
	for (int i=0; i<7; i++) {
		int bit = (kase>>i) & 1;
		
		if (bit) { // 编号为i的二极管点亮 
			// 看看是否能够与现有的点亮的二极管进行合并
			merge(i, k);	
			lighted[k++] = i;
		} 
	}
	
	// 检查是否只有一个连通块
	set<int> blockSet;
	for(int i=0; i<k; i++) {
		blockSet.insert(find( lighted[i] )); // 注意是 find(lighted[i])，而并非 par[lighted[i]] 
	} 
//	if (blockSet.size() == 1) {
//		for (int i=0; i<k; i++) {
//			printf("%d ", lighted[i]);
//		}
//		printf("\n");
//	}
	return blockSet.size()==1;
} 

int input() {
	int a, b;
	while (cin>>a>>b) {
		g[a][b] = g[b][a] = 1;
	}
}

int main() {
	
	input();
	printf("--\n");
	int ans = 0;
	int n = 1<<7;
//	printf("%d\n", isBlock(n-1));
	
	for(int i=1; i<n; i++) {
		ans += isBlock(i);
	}
	
	printf("%d\n", ans);
	
	return 0;
}