力扣面试150题--除法求值

Day 62

题目描述

做法

此题本质是一个图论问题,对于两个字母相除是否存在值,其实就是判断,从一个字母能否通过其他字母到达,做法如下:

  1. 遍历所有等式,为每个变量分配唯一的整数索引。
  2. 初始化一个二维数组 graph,其中 graph[i][j] 表示变量 i 除以变量 j 的比值。
  3. 对于每个等式 a/b = v,设置 graph[a][b] = v 和 graph[b][a] = 1/v。
  4. 每个变量自身的比值为 1.0,即 graph[i][i] = 1.0。
  5. 对于每个查询 a/b,检查变量是否存在于映射中。如果存在,使用 DFS 查找从 a 到 b 的路径,并计算路径上的比值乘积。如果路径不存在,返回 -1.0。
java 复制代码
class Solution {
    public double find(double[][] graph, int x, int y, boolean[] visited) {//判断x到y是否可达
        if (graph[x][y] != 0) {
            return graph[x][y];//直接可达
        }
        visited[x] = true;
        for (int i = 0; i < graph.length; i++) {
            if (graph[x][i] != 0 && !visited[i]) {
                double p = find(graph, i, y, visited);
                if (p != 0.0) {
                    return p * graph[x][i];
                }
            }
        }
        return 0;
    }

    public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {
        Map<String, Integer> num = new HashMap<>();//字符串到数字序号的转换
        int x = 0, y = 0;
        String a, b;
        for (int i = 0; i < equations.size(); i++) {
            a = equations.get(i).get(0);
            b = equations.get(i).get(1);
            if (!num.containsKey(a)) {
                num.put(a, x);
                x++;
            }
            if (!num.containsKey(b)) {
                num.put(b, x);
                x++;
            }
        }
        double[][] graph = new double[x][x];
        for (int i = 0; i < equations.size(); i++) {
            a = equations.get(i).get(0);
            b = equations.get(i).get(1);
            x = num.get(a);
            y = num.get(b);
            graph[x][y] = values[i];//x/y
            graph[y][x] = 1 / values[i];//y/x
        }
        for (int i = 0; i < graph.length; i++) {
            graph[i][i] = 1.0;//对角线全为1
        }
        //图初始化结束
        double[] res = new double[queries.size()]; //结果集合
        double sum = 0.0;
        for (int i = 0; i < queries.size(); i++) {
            a = queries.get(i).get(0);
            b = queries.get(i).get(1);
            if (!num.containsKey(a) || !num.containsKey(b)) {
                res[i] = -1.0;
                continue;
            }
            x = num.get(a);
            y = num.get(b);
            boolean[] visited = new boolean[graph.length];
            sum = find(graph, x, y, visited);//判断x到y是否可达
            if (sum == 0) {
                //不可达
                sum = -1.00000;
            }
            res[i] = sum;
            sum = 0.0;
        }
        return res;
    }
}
相关推荐
凌肖战21 分钟前
力扣网C语言编程题:在数组中查找目标值位置之二分查找法
c语言·算法·leetcode
weixin_4786897643 分钟前
十大排序算法汇总
java·算法·排序算法
luofeiju1 小时前
使用LU分解求解线性方程组
线性代数·算法
天涯学馆1 小时前
前端开发也能用 WebAssembly?这些场景超实用!
前端·javascript·面试
SKYDROID云卓小助手2 小时前
无人设备遥控器之自动调整编码技术篇
人工智能·嵌入式硬件·算法·自动化·信号处理
ysa0510302 小时前
数论基础知识和模板
数据结构·c++·笔记·算法
GEEK零零七2 小时前
Leetcode 1103. 分糖果 II
数学·算法·leetcode·等差数列
今天背单词了吗9802 小时前
算法学习笔记:7.Dijkstra 算法——从原理到实战,涵盖 LeetCode 与考研 408 例题
java·开发语言·数据结构·笔记·算法
重庆小透明4 小时前
力扣刷题记录【1】146.LRU缓存
java·后端·学习·算法·leetcode·缓存
desssq4 小时前
力扣:70. 爬楼梯
算法·leetcode·职场和发展