思路:每个数码管都有两种选择,亮或者不亮。其实就是{a, b, c, d, e, f, g}的所有子集,然后从所有子集中选出子集元素相通的,就是合法的。
先枚举所有子集,然后通过dfs看该子集元素是否相通。
代码如下:
import java.util.ArrayList;
import java.util.List;
public class D七段码 {
static int n = 7;
static int ans = 0;
static int[][] map = new int[n][n];
public static void main(String[] args) {
//建立关系图
map[0][1] = 1;
map[0][5] = 1;
map[1][0] = 1;
map[1][2] = 1;
map[1][6] = 1;
map[2][1] = 1;
map[2][3] = 1;
map[2][6] = 1;
map[3][2] = 1;
map[3][4] = 1;
map[4][3] = 1;
map[4][5] = 1;
map[4][6] = 1;
map[5][0] = 1;
map[5][4] = 1;
map[5][6] = 1;
map[6][1] = 1;
map[6][2] = 1;
map[6][4] = 1;
map[6][5] = 1;
List<Integer> path = new ArrayList<Integer>();
dfs(path, 0);
System.out.println(ans);
}
private static void dfs(List<Integer> path, int k) {
if(k == n) {
// System.out.println(path);
if(check(path)) {
ans++;
}
return;
}
dfs(path, k + 1);
path.add(k);
dfs(path, k + 1);
path.remove(path.size() - 1);
}
private static boolean check(List<Integer> path) {
if(path.size() == 1) {
return true;
}
List<Integer> temp = new ArrayList<>(path);
if(!temp.isEmpty()) {
f(temp, temp.get(0));
}
return temp.isEmpty();
}
private static void f(List<Integer> temp, int cur) {
//一个元素的时候, map[i][i] == 0, 自己跟自己没关系。
for(int i = 0; i < n; i++) {
if(temp.contains(i) && map[cur][i] == 1) {
temp.remove((Integer)i);
f(temp, i);
}
}
}
}
竞赛的时候,求出来的结果是54, 错在了两个地方
-
check()中应该返回
temp.isEmpty();
而不是!temp.isEmpty();
因为该子集中连通的元素被remove掉,如果该子集是连通的,那么temp为空,即该子集合法。 -
竞赛的时候,没写这个
if(path.size() == 1) { return true; }
我的f()找连通,有一点问题就是只有一个元素的时候,map[i][i] != 1,因此一个元素的情况没算上,结果就变少了。
总结:今年是第一次参加蓝桥杯,从寒假准备,后来有了疫情,一直在家,每天在leetcode上刷刷算法题,也算是在一直坚持。昨天发挥正常,遗憾的是深搜很熟练,但是check条件没写好,丢分了。总的来说,还算不错吧。