leetcode 399. Evaluate Division (medium)

题目链接

---

思路：

建立图结构，（节点可以用自定义结构体或类，也可以用容器配合pair）被除数/除数作为节点，之间的商作为节点间的权重值。
转换成图后，题目可以转换为求两点间的权重值。因为不需要寻求最短路径，所以dfs和bfs都可，这里使用dfs。

class Solution
{
public:
	unordered_map<string, vector<pair<string, double>>> mGraph;

	vector<double> calcEquation(vector<vector<string>> &equations, vector<double> &values, vector<vector<string>> &queries)
	{
		vector<double> res;
		// 建立图
		for (int i = 0; i < equations.size(); i++)
		{
			mGraph[equations[i][0]].push_back({equations[i][1], values[i]});
			mGraph[equations[i][1]].push_back({equations[i][0], 1 / values[i]});
		}
		// 求结果
		for (int i = 0; i < queries.size(); i++)
		{
			res.push_back(dfs(queries[i][0], queries[i][1]));
		}
		return res;
	}

	double dfs(const string &src, const string &des)
	{
		if (!mGraph.count(src) || !mGraph.count(des))
		{
			return -1.0;
		}
		if (src == des)
		{
			return 1.0;
		}

		queue<pair<string, double>> que;
		unordered_set<string> visited;

		que.push({src, 1.0});
		while (!que.empty())
		{
			pair<string, double> mSrcVal = que.front();
			que.pop();

			if (mSrcVal.first == des)
			{
				return mSrcVal.second;
			}

			visited.insert(mSrcVal.first);
			for (auto vec : mGraph[mSrcVal.first])
			{
				if (!visited.count(vec.first))
				{
					que.push({vec.first, vec.second * mSrcVal.second});
				}
			}
		}
		return -1;
	}
};