给出方程式 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<double>& values, vector<pair<string, string>> queries(方程式,方程式结果,问题方程式),
其中 equations.size() == values.size(),即方程式的长度与方程式结果长度相等(程式与结果一一对应),并且结果值均为正数。
以上为方程式的描述。 返回vector<double>类型。
基于上述例子,输入如下:
equations(方程式) = [ ["a", "b"], ["b", "c"] ],
values(方程式结果) = [2.0, 3.0],
queries(问题方程式) = [ ["a", "c"], ["b", "a"], ["a", "e"], ["a", "a"], ["x", "x"] ].
输入总是有效的。你可以假设除法运算中不会出现除数为0的情况,且不存在任何矛盾的结果。
思路分析:如示例中的a / b和b / c需要求解a / c,a / c == (a / b) * (b / c)。这个过程是不是有点类似于a到b,然后b到c,这不就是一个图的问题吗?
方法一:使用深度优先搜索法。
class Solution {
public:
set<string> points;//点集
vector<double> result;//储存结果
map<pair<string, string>, double> myMap;//myMap[{a,b}]代表a / b的值
vector<double> calcEquation(vector<pair<string, string>> equations, vector<double>& values, vector<pair<string, string>> queries) {
int equationsSize = equations.size();
//第一步:利用equations和values进行建图
for (int index = 0; index < equationsSize; ++index) {
//将点放入点集
points.insert(equations[index].first);
points.insert(equations[index].second);
//构建边的信息
myMap[equations[index]] = values[index];//正向
myMap[{
equations[index].second, equations[index].first}] = 1 / values[index