原题
小蓝要用七段码数码管来表示一种特殊的文字。上图给出了七段码数码管的一个图示,数码管中一共有 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判断图是否连通
- 从某个结点 v 出发,访问结点 v,并令 vis[v] = 1
- 查找 v 的所有邻接点 i,若结点 i 并未被访问过(vis[i] = 0),则从结点 i 出发,继续深搜,转至步骤1
- 深搜完整个图之后,遍历 vis 数组。如果数组全是1,说明只有一个连通分量,所有节点同属一个连通块。否则就存在多个连通分量
并查集判断图是否连通
默认你已经学习过并查集
- 初始化每个结点的根节点为结点本身(开始时各个节点分别属于不同集合),即:parent[i]=i
- 从某个结点 v 开始,查找 v 的邻接点 i,如果 v 的根节点和 i 的根节点不同(节点 v 和节点 i 属于不同集合),则将 i 的根节点指向 v 的根节点(或者将v 的根节点指向 i 的根节点)(合并两个集合)
- 遍历完整个图后,如果所有节点的根节点都一样,则表示所有节点属于同一个集合(连通分量、连通块)。否则就存在多个连通分量
代码
代码中有注释,有些细节处的代码需要注意和思考!
代码中也有一些为了调试,添加的一些打印代码。
个人比较推荐: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;
}