Equations are given in the format A / B = k
, where A
and B
are variables represented as strings, and k
is a real number (floating point number). Given some queries, return the answers. If the answer does not exist, return -1.0
.
Example:
Given a / b = 2.0, b / c = 3.0.
queries are: a / c = ?, b / a = ?, a / e = ?, a / a = ?, x / x = ? .
return [6.0, 0.5, -1.0, 1.0, -1.0 ].
The input is: vector<pair<string, string>> equations, vector<double>& values, vector<pair<string, string>> queries
, where equations.size() == values.size()
, and the values are positive. This represents the equations. Return vector<double>
.
According to the example above:
equations = [ ["a", "b"], ["b", "c"] ], values = [2.0, 3.0], queries = [ ["a", "c"], ["b", "a"], ["a", "e"], ["a", "a"], ["x", "x"] ].
The input is always valid. You may assume that evaluating the queries will result in no division by zero and there is no contradiction.
数据结构:用Map<String, Map<String, double>>来存储dividend- divisor- quotient关系。每读到一个信息的时候需要加入4行数据:dividend-divisor-quotient, dividend-dividend-1, divisor-dividend-1/quotient, divisor-divisor-1
算法:DFS。这个宏观大问题拆解成小问题时,变的其实是被除数,不变的是最终的除数,试着找一条链条最后导向最终除数。函数头private double dfs(Map<String, Map<String, Double>> map, String dividend, String finalDivisor, Set<String> visited) 。利用map,找到dividend / finalDivisor的答案并返回,同时找的过程中利用visited避免循环查找导致stackOverFlow。
1.如果直接没有dividend,肯定失败。
2.在dividend对应的keySet()也就是divisor里循环。
2.1如果找到finalDivisor了,整个dfs都结束了,一层层返回
2.2退而求其次把这个divisor当成下一层的dividend递归,试着递归到链条传到finalDivisor。
2.3如果这个divisor已经被visited了,肯定没必要再找了,continue
2.4递归一下(上下状态修改)
2.5递归成功的话帮助传上去
3.全循环完了也没有在中间return回合格结果的话,那肯定失败了,传回失败标志。
细节:1.建关系映射的时候要记得加入自除和反除信息,一共四条。2.要使用Set visited记录找过的不可能的那些被除数避免重复工作,有时候重复还会危险到循环查找导致stackOverFlow。set的操作对象都是divisor。因为你接下来要做的工作是把divisor当成新的被除数有没有希望,所以你要做的是1.检查这个divisor被做过没有,2.改变状态记录Set里的divisor状态 3.递归divisor。这些作用目标应该有一致性,而不是去操作当前的dividend。
实现:
class Solution { public double[] calcEquation(String[][] equations, double[] values, String[][] queries) { Map<String, Map<String, Double>> map = new HashMap<>(); for (int i = 0; i < equations.length; i++) { String dividend = equations[i][0]; String divisor = equations[i][1]; double quotient = values[i]; if (!map.containsKey(dividend)) { map.put(dividend, new HashMap<String, Double>()); } if (!map.containsKey(divisor)) { map.put(divisor, new HashMap<String, Double>()); } // 有一个自除千万记得加! map.get(dividend).put(dividend, 1.0); map.get(divisor).put(divisor, 1.0); // 有一个反除千万记得加! map.get(dividend).put(divisor, quotient); map.get(divisor).put(dividend, 1 / quotient); } double[] result = new double[queries.length]; for (int i = 0; i < queries.length; i++) { result[i] = dfs(map, queries[i][0], queries[i][1], new HashSet<String>()); } return result; } private double dfs(Map<String, Map<String, Double>> map, String dividend, String finalDivisor, Set<String> visited) { if (!map.containsKey(dividend)) { return -1.0; } for (String divisor : map.get(dividend).keySet()) { double quotient = map.get(dividend).get(divisor); if (divisor.equals(finalDivisor)) { return quotient; } // 注意用set。避免同一个被除数被反复检查除以finalDivisor导致stackoverflow if (visited.contains(divisor)) { continue; } visited.add(divisor); double deepQ = dfs(map, divisor, finalDivisor, visited); visited.remove(divisor); if (deepQ != -1.0) { return quotient * deepQ; } } return -1.0; } }