leetcode 第399题 除法求值

题目描述：
给出方程式 A / B = k, 其中 A 和 B 均为用字符串表示的变量， k 是一个浮点型数字。根据已知方程式求解问题，并返回计算结果。如果结果不存在，则返回 -1.0。

示例 :
给定 a / b = 2.0, b / c = 3.0
问题: a / c = ?, b / a = ?, a / e = ?, a / a = ?, x / x = ?
返回 [6.0, 0.5, -1.0, 1.0, -1.0 ]
输入为: vector<pair<string, string>> equations, vector& values, vector<pair<string, string>> queries(方程式，方程式结果，问题方程式)， 其中 equations.size() == values.size()，即方程式的长度与方程式结果长度相等（程式与结果一一对应），并且结果值均为正数。以上为方程式的描述。 返回vector类型。
基于上述例子，输入如下：
equations(方程式) = [ [“a”, “b”], [“b”, “c”] ],
values(方程式结果) = [2.0, 3.0],
queries(问题方程式) = [ [“a”, “c”], [“b”, “a”], [“a”, “e”], [“a”, “a”], [“x”, “x”] ].
输入总是有效的。你可以假设除法运算中不会出现除数为0的情况，且不存在任何矛盾的结果。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/evaluate-division
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解：

class Solution 
{
public:
unordered_map<string, vector<pair<string, double>>> ump;
	double getdist(string &a, string &b)
	{
		//没出现过直接返回-1.0
		auto it1 = ump.find(a);
		auto it2 = ump.find(b);
		if (it1 == ump.end() || it2 == ump.end())
			return -1.0;

		//相等直接返回1.0
		if (a == b)
			return 1.0;

		//BFS遍历
		queue<pair<string, double>> Q;
		Q.push({ a, 1.0 });
		unordered_map<string, int> vis;
		vis[a] = 1;

		double ret = -1.0;
		while (!Q.empty())
		{
			auto f = Q.front();
			Q.pop();

			//记录为已访问
			vis[f.first] = 1;
			//已到达目标，返回当前值
			if (f.first == b)
			{
				ret = f.second;
				break;
			}
			for (auto &pr : ump[f.first])
			{
				//新节点的值乘上权重
				if (vis[pr.first] == 0)
					Q.push({ pr.first, pr.second*f.second });
			}
		}
		return ret;
	}
    vector<double> calcEquation(vector<vector<string>>& equations, vector<double>& values, vector<vector<string>>& queries) 
    {
        int n = equations.size();
		while (n--)
		{
			//建立双向图
			ump[equations[n][0]].push_back({ equations[n][1], values[n] });
			ump[equations[n][1]].push_back({ equations[n][0], 1.0 / values[n] });
		}

		vector<double> ret;
		for (auto &vec : queries)
			ret.push_back(getdist(vec[0], vec[1]));
		return ret;
    }
};

解题思路：
建立一个双向表，存储着一个数到另一个数的乘积，相当于权值。
利用这个双向表，从被除数开始，利用BFS，每次将自己的权值乘到与它相连的节点的权值上，然后入队列，直到到除数为止。