除法求值

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）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。