算法竞赛入门经典 例题6-19

UVa1572

Self-Assembly

假设分子为四边形，每条边用一个字母和符号标识。当两个分子的两条边字母相同且符号不同时，两个分子可以连接在一起。给定一些分子，判断这些分子是否可以无限连接。

根据题目中的图片，首先想到将分子视为顶点，然后根据每条边的字母和符号添加图中的边，最后只需要判断图中是否存在环即可。这种方法不仅难写，而且超时了，原因在于分子的种类高达40000。

所以要换一种建模方法，也就是将字母和符号的组合视为顶点，这样顶点数量只有52个。添加边的方式为：如果分子表示为A+00A+A+，也就意味着另一个分子的A-边可以连接到这个分子。最终还是判断图中是否存在环。

好久不写深搜判环，调试了好久 😦😦😦

#include <iostream>
#include <array>
#include <vector>
#include <string>

using namespace std;

class Solution
{
public:
	Solution(const vector<string> &molecules) 
		: graph(52, vector<bool>(52, false)), visited(52, false), circle(false)
	{
		for (const string &molecule : molecules)
		{
			array<char, 4> labels;
			for (size_t i = 0; i < molecule.length(); i += 2)
			{
				if (molecule[i] == '0' && molecule[i + 1] == '0') {
					labels[i >> 1] = '0';
				}
				else if (molecule[i + 1] == '+') {
					labels[i >> 1] = toupper(molecule[i]);
				}
				else{
					labels[i >> 1] = tolower(molecule[i]);
				}
			}
			for (size_t i = 0; i < labels.size(); i++)
			{
				if (labels[i] == '0') continue;
				char adj = GetConnected(labels.at(i));
				for (size_t j = 0; j < labels.size(); j++)
				{
					if (i == j || labels[j] == '0') continue;
					graph[Char2Index(adj)][Char2Index(labels.at(j))] = true;
				}
			}
		}
		for (size_t i = 0; i < graph.size(); i++)
		{
			visited[i] = true;
			SearchCircle(i);
			visited[i] = false;
			if (circle) break;
		}
	}
	bool HasCircle()
	{
		return circle;
	}
private:
	vector<vector<bool>> graph;
	vector<bool> visited;
	bool circle;
	int Char2Index(char ch)
	{
		if (isupper(ch)) return ch - 'A';
		else return ch - 'a' + 26;
	}
	char GetConnected(char ch)
	{
		if (isupper(ch)) return tolower(ch);
		else return toupper(ch);
	}
	void SearchCircle(size_t current)
	{
		for (size_t i = 0; i < graph[current].size(); i++)
		{
			if (!graph[current][i]) continue;
			if (visited[i]) {
				circle = true;
			}
			else {
				visited[i] = true;
				SearchCircle(i);
				visited[i] = false;
			}
			if (circle) break;
		}
	}
};

int main()
{
	size_t n = 0;
	while (cin >> n) {
		string molecule;
		vector<string> molecules;
		for (size_t i = 0; i < n; i++)
		{
			cin >> molecule;
			molecules.push_back(molecule);
		}
		Solution solution(molecules);
		if (solution.HasCircle()) {
			cout << "unbounded" << endl;
		}
		else {
			cout << "bounded" << endl;
		}
	}
	return 0;
}
/*
3
A+00A+A+ 00B+D+A- B-C+00C+
1
K+K-Q+Q-
*/