LeetCode
除法求值
给出方程式 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的情况,且不存在任何矛盾的结果。
解法:并查集
解题思路:
我们使用并查集来保存每一个除数和被除数的关系
这里我的并查集的方向是
a/b=2 , b/c=3
c 1
^
|
b 3
^
|
a 2
val代表的是当前节点是直接父节点的多少倍,也就是 A/fatherA ,重点就是合并两个集合的时候需要注意:
已知
a / fa = val[a]
b / fb = val[b]
现在我们要合并a,b且 a / b=value
所以我们需要设置 parent[fa]=fb
由于fa父节点发生了变化所以它的值也需要变化,也就是要求 fa/fb的值
val[fa] = fa/fb = a/b * b/fb * fa/a = value * (val[b] / val[a])
代码如下:
class Solution{
private HashMap<String,String> parent=new HashMap<>();
private HashMap<String,Double> quotient=new HashMap<>();
//不带路径压缩
public String find(String p){ //找根结点
while (parent.get(p)!=p) {
p=parent.get(p);
}
return p;
}
public void init(String s){ //初始化
if (!parent.containsKey(s)) {
parent.put(s,s);
quotient.put(s,1.0);
}
}
public void merge(String a,String b,Double value){
init(a);init(b);
String fa=find(a); // a/fa=val[a], b/fb=val[b]
String fb=find(b);
if (fa.equals(fb)) {
return;
}
parent.put(fa,fb);
quotient.put(fa,value*(cal(b)/cal(a))); //cal(a)和cal(b)代表a和b到根节点的总值
}
public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {
for (int i=0;i<equations.size();i++) {
List<String> equation=equations.get(i);
merge(equation.get(0),equation.get(1),values[i]);
}
double[] res=new double[queries.size()];
int index=0;
for (List<String> query:queries) {
String a=find(query.get(0));
String b=find(query.get(1));
System.out.println(a+" "+b);
if (!parent.containsKey(query.get(0)) || !parent.containsKey(query.get(1)) || !a.equals(b)) {
res[index++]=-1;
}else{
//没有路劲压缩,需要遍历整个路劲求积
res[index++]=cal(query.get(0))/cal(query.get(1));
}
}
return res;
}
//计算当前节点到根节点的路径乘积
public double cal(String index){
double res=quotient.get(index);
while(parent.get(index)!=index){
index=parent.get(index);
res*=quotient.get(index);
}
return res;
}
}
解法来源
作者:resolmi
链接:https://leetcode-cn.com/problems/evaluate-division/solution/javabing-cha-ji-lu-jing-ya-suo-by-im1gw0/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。