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的情况,且不存在任何矛盾的结果。
PS:
用map构建关系,然后dfs找
class Solution {
public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {
Map<String, Map<String,Double>> graph = doGraph(equations, values);
double[] res = new double[queries.size()];
int index = 0;
for(List<String> q : queries){
res[index++] = dfs(graph, new HashSet<>(), q.get(0), q.get(1), 1);
}
return res;
}
public double dfs(Map<String, Map<String,Double>> graph, Set<String> visited, String start, String end, double ans){
if(!graph.containsKey(start) || !graph.containsKey(end)) return -1;
Map<String, Double> edges = graph.get(start);
for(String key : edges.keySet()){
if(!visited.contains(key)){
visited.add(key);
double v = edges.get(key);
if(key.equals(end)) return ans*v; //
double d = dfs(graph, visited, key, end, ans*v);
if(d != -1) return d;
}
}
return -1;
}
public Map<String, Map<String,Double>> doGraph(List<List<String>> equations, double[] values){
Map<String, Map<String,Double>> graph = new HashMap<>();
for(int i = 0 ; i < equations.size(); i++){
String s = equations.get(i).get(0);
String t = equations.get(i).get(1);
double val = values[i];
Map<String, Double> edge1 = graph.getOrDefault(s, new HashMap<>());
edge1.put(t, val);
graph.put(s, edge1);
Map<String, Double> edge2 = graph.getOrDefault(t, new HashMap<>());
edge2.put(s, 1/val);
graph.put(t, edge2);
}
return graph;
}
}