LeetCode0399.除法求值 Go语言AC笔记

时间复杂度:O(n²)

解题思路

该题可以用图的方法去做，也可以用并查集去做，更推荐后者，所以只讲并查集解题思路。

如果是A/B=3，那么就令A的father是B，也就是把除数看成是被除数的father，father[A]=B=father[B]。

我们还需要一个权值数组来记录每个字符串除以它的father的商，比如weight[A]=3，weight[B]=1。

我们再加入一个B/C=2，这样father[B]就是C，在查找A的father时，father[A]=B→father[B]=C→father[C]=C，现在C的father是自身，所以A的father就是C了。但是我们的weight数组不要忘记做修改，需要乘上weight[B]，现在B的father是C，所以weight[B]=B/C=2而不是原来的1，weight[A]=A/C=A/B*B/C=weight[A]*weight[B]，所以在查找father 的过程中weight要更新，乘上weight[旧father]即可。

在添加一个新式子需要合并father时，需要将被除数的father合并到除数的father（千万不要弄反了），所以我们的weight[除数的旧father]需要更改，那就是 新式子的商×weight[被除数]/weight[除数]。

以上是并查集的设计思路。

我们需要将每个字符串记录下来，这样在遇到没有的字符串时直接返回-1，所以我们用map来记录已有的字符串，value就是从0累加的id。

当所有已知式子merge到并查集中后，我们要对待求式子进行判断。首先看是否有这两个字符串，如果有就继续判断二者是否是同一个father，如果是说明二者有关系可以进行除法计算，否则两个字符串属于不同的集合无法进行除法计算，故返回-1。

AC代码

func calcEquation(equations [][]string, values []float64, queries [][]string) []float64 {
    id:=map[string]int{}
    for _,v:=range equations{
        if _,ok:=id[v[0]];!ok{//不在id中
            id[v[0]]=len(id)//id就是当前的map长度
        }
        if _,ok:=id[v[1]];!ok{
            id[v[1]]=len(id)
        }
    }
    father:=make([]int,len(id))//记录每个字符串的father，即除数
    weight:=make([]float64,len(id))//记录与father当除数时的商
    for i:=0;i<len(father);i++{
        father[i]=i//初始化father为自己
        weight[i]=1//因为father为自己故商为1
    }
    var find func(int)int
    find=func(x int)int{
        if father[x]!=x{
            fa:=find(father[x])
            weight[x]*=weight[father[x]]//寻找father本身的过程就是不断找除数的过程，所以x除以新father的商需要乘上老father除以新father 的商
            father[x]=fa
        }
        return father[x]//x==father[x]
    }
    merge:=func(left,right int,val float64){
        leftFather,rightFather:=find(left),find(right)
        weight[leftFather]=val*weight[right]/weight[left]//left老father除以新father的商
        father[leftFather]=rightFather//令left老father的新father为right的father
    }
    for k,v:=range equations{
        merge(id[v[0]],id[v[1]],values[k])
    }
    res:=make([]float64,len(queries))
    for i,query:=range queries{
        left,hasLeft :=id[query[0]]
        right,hasRight:=id[query[1]]
        if hasLeft&&hasRight&&find(left)==find(right){
            res[i]=weight[left]/weight[right]
        }else{
            res[i]=-1
        }
    }
    return res
}

感悟

再一次要吐，一道medium又做出了hard的感觉。是不是21的medium难度升级了？