【Leetcode】399. Evaluate Division

题目地址：

https://leetcode.com/problems/evaluate-division/

给定若干字符串对组成的列表$A$，和一个double数组$v$，将这些字符串看成是变量，则有$A[i][0]/A[i][1]=v[i]$。再给定若干询问，问两个字符串之商。若这两个字符串其中一个不存在，或商求不出来，则输出$-1$。题目保证$v$是正数数组。

可以用类似Floyd算法的方法求任意两个字符串的商，然后应答询问。为了提升效率，可以将字符串映射为一个下标，然后开double数组做Floyd算法。代码如下：

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Solution {
    public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {
        Map<String, Integer> map = new HashMap<>();
        int size = 0;
        for (List<String> equation : equations) {
            // 注意不要写成map.putIfAbsent(equation.get(0), size++);，这样写的话size会不停加，即使没put，会浪费空间
            if (!map.containsKey(equation.get(0))) {
                map.put(equation.get(0), size++);
            }
            if (!map.containsKey(equation.get(1))) {
                map.put(equation.get(1), size++);
            }
        }
        
        double[][] mat = new double[size][size];
        for (int i = 0; i < equations.size(); i++) {
            String s1 = equations.get(i).get(0), s2 = equations.get(i).get(1);
            int x = map.get(s1), y = map.get(s2);
            mat[x][x] = mat[y][y] = 1.0;
            mat[x][y] = values[i];
            mat[y][x] = 1 / values[i];
        }
    
    	// k是枚举走的中间点的下标范围，范围是0 ~ k
        for (int k = 0; k < size; k++) {
        	// 枚举路径的两个端点
            for (int i = 0; i < size; i++) {
                for (int j = 0; j < size; j++) {
                    if (mat[i][k] != 0 && mat[k][j] != 0) {
                        mat[i][j] = mat[i][k] * mat[k][j];
                    }
                }
            }
        }
        
        double[] res = new double[queries.size()];
        for (int i = 0; i < queries.size(); i++) {
            String s1 = queries.get(i).get(0), s2 = queries.get(i).get(1);
            int x = map.getOrDefault(s1, -1), y = map.getOrDefault(s2, -1);
            res[i] = x == -1 || y == -1 || mat[x][y] == 0 ? -1 : mat[x][y];
        }
        
        return res;
    }
}

预处理时间$O(n^3+n^{2}l)$，$n$是不同字符串数量，$l$是最长字符串长度，每次询问时间$O(1)$，空间$O(nl+n^2)$。