除法求值-并查集-Java实现

一开始看除法求值题是想用图来做，结果看了半天别人的答案，看半天都没看出来哪有图，去搜了搜并查集，啊～～

问题：

给出方程式 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的情况，且不存在任何矛盾的结果。

//当前节点的父节点
HashMap<String, String> parent = new HashMap<>();

//当前节点是父节点的n倍，n = 父 / 子
HashMap<String, Double> quotient = new HashMap<>();

public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {
    int n = equations.size();
    for (int i = 0; i < n; i++) {
        String s0 = equations.get(i).get(0);
        String s1 = equations.get(i).get(1);
        double value = values[i];
        merge(s0, s1, value);
    }
    n = queries.size();
    double[] res = new double[n];
    for (int i = 0; i < n; i++) {
        List<String> list = queries.get(i);
        String s0 = list.get(0);
        String s1 = list.get(1);
        String r0 = findRoot(s0);
        String r1 = findRoot(s1);
        if (r0 != null && r0.equals(r1)) {
            res[i] = call(s1) / call(s0);
        } else {
            res[i] = -1;
        }
    }
    return res;
}

private void merge(String s0, String s1, double value) {
    init(s0);
    init(s1);
    String r0 = findRoot(s0);
    String r1 = findRoot(s1);
    if (r0.equals(r1)) {
        return;
    }
    parent.put(r1, r0);
    quotient.put(r1, value * call(s0) / call(s1));
}

private void init(String s) {
    if (!parent.containsKey(s)) {
        parent.put(s, s);
        quotient.put(s, 1.0);
    }
}

private String findRoot(String x) {
    String p;
    while ((p = parent.get(x)) != null && !p.equals(x)) {
        x = p;
    }
    return p;
}

private double call(String s) {
    double res = 1;
    while (!parent.get(s).equals(s)) {
        res *= quotient.get(s);
        s = parent.get(s);
    }
    return res;
}