蓝桥杯——七段码（并查集+二进制情况罗列）

• 问题网站

  https://www.lanqiao.cn/problems/595/learning/?contest_id=73

• 这道题就是相邻的段可以表示一种符号，最少必须要有一段。其实我最初的想法就是把全部的符号表示按照符号个数分别罗列出来，但是中间过程肯定会有遗漏。

• 但是解法采用的是并查集+二进制生成的思想，利用递归的思想总共7个灯，表示一个二进制数，为1表示灯量，为0表示灯暗，让灯亮的所有段做并查处理，如果最后各个段的祖先段都相同，说明这些亮着的段都是相连的，符合符号的表示规则。

• 二进制生成的代码如下，采用递归的思想。(这个思想最好理解并且记忆下来)

// 初始use[d] = 0
// 按照递增的顺序生成7位二进制的所有可能 0000000 —— 1111111
void dfs(int d){
	if(d > 7){
        for (int i = 1; i <= 7;i++)
        {
            cout << use[i] << " ";
        }
	}
	// 数组从1到7表示从高位到低位。
    // use[d]=1后面直接进入函数，让d位置后面的数全部被置为0，只有到d=7时候实现一次0到1的变换才返回。让前面的调用处从0变成1。
    // 记住这个递归产生二进制的代码，很有用。
	use[d] = 0;//打开d这个灯，继续开关下一个灯
    
    dfs(d + 1);
	use[d] = 1;//关闭d这个灯，继续开关下一个灯
	dfs(d + 1);
}

• 整个题目的代码

#include <iostream>
using namespace std;
int fa[10];
int use[10];
int A[10][10];
int ans;
const int N = 7;

// 初始化哪些段之间有连接
void init()
{
    A[1][2] = 1;
    A[1][6] = 1;
    A[2][1] = 1;
    A[2][3] = 1;
    A[2][7] = 1;
    A[3][2] = 1;
    A[3][4] = 1;
    A[3][7] = 1;
    A[4][3] = 1;
    A[4][5] = 1;
    A[5][4] = 1;
    A[5][6] = 1;
    A[5][7] = 1;
    A[6][1] = 1;
    A[6][7] = 1;
    A[6][5] = 1;
    A[7][2] = 1;
    A[7][3] = 1;
    A[7][5] = 1;
    A[7][6] = 1;
}

// 找到点的祖先连接点
int getTop(int x)
{
    if(fa[x] == x)
    {
        return x;
    }
    fa[x] = getTop(fa[x]);
    return fa[x];
}

// 合并两个有联系的点
void Merge(int x,int y)
{
    int top1 = getTop(x);
    int top2 = getTop(y);
    if(top1!=top2)
    {
        fa[top1] = top2;
    }
}

// 罗列二进制位从0000000 -- 1111111（逐一递增）
void dfs(int d)
{
    // 说明此时产生了一个新的二进制序列
    if(d>7)
    {
        // 并查集第一步，需要将连接数组置位本身
        for (int i = 1; i <= N;i++)
        {
            fa[i] = i;
        }
        // 开始进行并与查的操作，这里面操作的对象必须是要亮灯的。
        for (int i = 1; i <= N;i++)
        {
            for (int j = i + 1; j <= N;j++)
            {
                if(A[i][j]==1 && use[i]==1 &&use[j]==1)
                {
                    Merge(i, j);
                }
            }
        }
        
        // 如果最后都为同一点，那么fa数组的值肯定都为相同的数。
        // 所以下面执行完k=1表示全部点的祖先结点相同。
        int k = 0;
        for (int i = 1; i <= N;i++)
        {   
            if(use[i]==1 && getTop(i)==i)
            {
                k++;
            }
        }
        if(k==1)
        {
            ans++;
        }
        return;
    }

    use[d] = 0;
    dfs(d + 1);
    use[d] = 1;
    dfs(d + 1);
}

int main()
{
    init();
    dfs(1);
    cout << ans << endl;
    return 0;
}