题目地址:
https://leetcode.com/problems/evaluate-division/
给定若干字符串对组成的列表 A A A,和一个double数组 v v v,将这些字符串看成是变量,则有 A [ i ] [ 0 ] / A [ i ] [ 1 ] = v [ i ] A[i][0]/A[i][1]=v[i] A[i][0]/A[i][1]=v[i]。再给定若干询问,问两个字符串之商。若这两个字符串其中一个不存在,或商求不出来,则输出 − 1 -1 −1。题目保证 v v 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 ) O(n^3+n^2l) O(n3+n2l), n n n是不同字符串数量, l l l是最长字符串长度,每次询问时间 O ( 1 ) O(1) O(1),空间 O ( n l + n 2 ) O(nl+n^2) O(nl+n2)。