初见此题没有思绪,我隐隐的感觉要用到并查集,但我又不知道怎么做,这个题比最简单的并查集要多一个难点,就是指定的方向才能将两个点链接,再一个就是这七个数码管亮和不亮的全排列怎么枚举。
第一个是我们数码管的全排列,一共是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()
{
src[0][1]=src[1][0]=1;
src[1][6]=src[6][1]=1;
src[5][6]=src[6][5]=1;
src[0][5]=src[5][0]=1;
src[2][6]=src[6][2]=1;
src[4][6]=src[6][4]=1;
src[2][1]=src[1][2]=1;
src[4][5]=src[5][4]=1;
src[4][3]=src[3][4]=1;
src[2][3]=src[3][2]=1;
int sum=0;
for(int i=0;i<128;i++){
int t=i;
for(int j=0;j<7;j++){
que[j]= (t&1);
t= t>>1 ;
}
if(check()){
sum++;
}
}
cout<<sum<<endl;
return 0;
}