2020第十一届蓝桥杯C/C++ 省赛A组 D题

2020第十一届蓝桥杯C/C++ 省赛A组 D题

思路：这题是深度优先遍历和并查集连通块的问题。首先需要用深度优先遍历每一种情况，在每一种情况中，利用并查集探查亮着的灯是否都连在一起（即是否连通块只有一个），当属于这种情况时，就符合规范，字符加一

#include<iostream>
#include<string>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;
const int N = 8;
int map[N][N], fa[N] , vis[N], ans;


int getfa(int x)
{
	if (fa[x] == x)
		return x;
	return getfa(fa[x]);
}

void dfs(int k)//深度优先遍历
{
	if (k > 7)//灯都枚举完了
	{
		for(int i=1;i<=7;i++)//并查集，先设自己为自己的父节点
		{
			fa[i] = i;
		}
		for (int i = 1; i <= 7; i++)
		{
			
			for (int j = 1; j <= 7; j++)
			{
				if (map[i][j] == 1 && vis[i] && vis[j])//相邻的灯亮的灯连在一起，形成连通块
				{
					int faa = getfa(i);
					int fab = getfa(j);
					if (faa != fab)
						fa[faa] = fab;
				}
			}
		}
		int cnt = 0;
		for (int i = 1; i <= 7; i++)
		{
			if (fa[i] == i && vis[i])//清点这种情况下连通块的数量
				cnt++;
		}
		if (cnt == 1)//连通块数量为1，说明所有亮灯都连在一起，符合
			ans++;
		return;//注意深度优先每次的出口要写出来
	}
	vis[k] = 1;//灯亮或者灯灭的遍历，遇到k>7才执行上面的内容
	dfs(k + 1);
	vis[k] = 0;
	dfs(k + 1);
}
int main()
{
	map[1][2] = map[2][1] = 1; map[1][6] = map[6][1] = 1;	//连通的存图 
	map[6][7] = map[7][6] = 1; map[6][5] = map[5][6] = 1;
	map[2][7] = map[7][2] = 1; map[2][3] = map[3][2] = 1;
	map[5][7] = map[7][5] = 1; map[5][4] = map[4][5] = 1;
	map[3][7] = map[7][3] = 1; map[4][3] = map[3][4] = 1;
	dfs(1);
	cout << ans;
	system("pause");
	return 0;
	
}