蓝桥杯day23_七段码——并查集&&dfs

思考

1：首先我们需要遍历所有情况，因此使用深搜dfs。请与全排列区分开，虽然都是用dfs 目前作者的初步判断是：全排列需要回溯（即状态恢复）；灯有开与不开两种情况（分而治之），因此状态恢复后也要进行遍历。
可以想象：
全排列就是闪灵里面小男主迷宫走到死路，就会原路返回，且不留痕迹（防止被爸爸追上）。
遍历所有情况则是题目中的LED灯，黑与亮都是它的一部分

2：排列结束后需要判断亮灯的部分是否是一个整体，这需要并查集的思想：如果两个 个体 相邻且均亮灯，那么他们便是一个整体，遍历所有个体，便可以把它们全部化成一个大整体
并查集的思想，可以单独查阅，大拿讲的清楚！

上代码！！！

#include <bits/stdc++.h>
using namespace std;
const int N = 10;
int pre[N];  //前驱节点
int maap[N][N];//是否联通
bool light[N]; //存储灯泡  亮   或者  不亮的状态
int ans; //答案，默认为0
void init(){                //并查集：初始化操作，不然你咋知道两个相邻不相邻？
    maap[1][2] = maap[1][6] = 1;
    maap[2][1] = maap[2][3] = maap[2][7] = 1;
    maap[3][2] = maap[3][7] = maap[3][4] = 1;
    maap[4][3] = maap[4][5] = 1;
    maap[5][6] = maap[5][7] = maap[5][4] = 1;
    maap[6][5] = maap[6][7] = maap[6][1] = 1;
    maap[7][6] = maap[7][2] = maap[7][3] = maap[7][5] = 1;
    for(int i = 1; i<=7; i++)       //并查集：一开始每个人都是自己的“老大”
        pre[i] = i;
}

int Find(int x){    //并查集：查“老大”&&优化路径
    if(pre[x] == x)
        return x;
    return pre[x] = Find(pre[x]);
}

void join(int x,int y){ //并查集：归并操作
    int fx = Find(x);
    int fy = Find(y);
    if(fx != fy)
        pre[fx] = fy;
}

void dfs(int n){
    if(n>7){
        init(); //每次初始化，觉得两点联通初始化次数多了，pre初始化额外放一个函数即可
        for(int i = 1; i<=7; i++){
            for(int j = 1; j<=7; j++){
                if(maap[i][j] && light[i] && light[j])
                    join(i,j);
            }
        }
        int x = 0;
        for(int i = 1; i<=7; i++)
            if(light[i] && pre[i]==i)
                x++;
        if(x == 1)      //如果只有一个且亮的整体，那么满足题意要求
            ans++;
        return ;
    }
    light[n] = 1;
    dfs(n+1);
    light[n] = 0;
    dfs(n+1);
}

int main(){
    dfs(1);
    cout<<ans;
}