七段码 蓝桥杯

初见此题没有思绪，我隐隐的感觉要用到并查集，但我又不知道怎么做，这个题比最简单的并查集要多一个难点，就是指定的方向才能将两个点链接，再一个就是这七个数码管亮和不亮的全排列怎么枚举。

第一个是我们数码管的全排列，一共是128种情况（包括全不亮），所以对应数字的二进制位的0和1可以一一对应每个数码管的亮或者不亮，我们利用位运算可以解决这个问题。

这个特定要求的并查集，我们可以设置二位数组，在判断两个发光管能否合并时可以使用，二位数组下标0-6分别代表a-g，如果这个数字是1则代表可以联通。

最后值得关注的是这个题目最后的答案里不包括灯管全不亮这个情况。记得排除。

#include<bits/stdc++.h>  //不亮的时候不算一个状态 
using namespace std;
int src[10][10];
int que[7]; 
int father[7];
int ifind(int x){//寻找其父节点 
	while(father[x]!=x){
		x=father[x];
	}
	return x;
}
bool check(){
	for(int i=0;i<7;i++) father[i]=i;
	
	for(int i=0;i<7;i++){
		for(int j=0;j<7;j++){
			if(que[i]&&que[j]&&src[i][j]){
				int t1=ifind(i);
				int t2=ifind(j);
				if(t1!=t2){
					father[t1]=t2;
				}
			}
		}
	}
	int sum=0;//计数有多少个联通集 
	for(int i=0;i<7;i++){
		if(que[i]&&(father[i]==i)){
			sum++;
		}
	}
	if(sum==1) return true;//只有一个联通集则符合条件 
	return false;
}
int main()
{
	//a b c d e f g 
	//0 1 2 3 4 5 6
	src[0][1]=src[1][0]=1;//a-b 
	src[1][6]=src[6][1]=1;//b-g
	src[5][6]=src[6][5]=1;//f-g
	src[0][5]=src[5][0]=1;//a-f
	src[2][6]=src[6][2]=1;//c-g
	src[4][6]=src[6][4]=1;//e-g
	src[2][1]=src[1][2]=1;//c-b
	src[4][5]=src[5][4]=1;//e-f
	src[4][3]=src[3][4]=1;//e-d
	src[2][3]=src[3][2]=1;//c-d
	
	int sum=0;
	for(int i=0;i<128;i++){
		int t=i;
		//cout<<t<<endl;
		for(int j=0;j<7;j++){
			que[j]= (t&1);
			t= t>>1 ;
		}
		if(check()){//检查此时是否合法 
			sum++;
		}
		
	}
	cout<<sum<<endl;
	
    return 0;
}