399. 除法求值（高频题）

末蓝、 2023-10-04 08:49 54阅读 0赞

#### 399. 除法求值 ####

*  解题思路
 *  代码

## 解题思路 ##

一刷2021/3/20，没弄明白，题解见[这里][Link 1]

## 代码 ##

// 定义邻接节点类
    class Node {
        
        // 邻接节点代表的字符串
        public String id;
        // 到达邻接节点所需的倍数
        public double num;
        public Node(String i, double n) {
        
            id = i;
            num = n;
        }
    }
    
    class Solution {
        
        // 构造一个map用于存储图
        Map<String, List<Node>> map;
        public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {
        
            int n = equations.size();
            map = new HashMap<>();
            double[] res = new double[queries.size()];
    
            // 将节点都放到图中
            for(int i = 0; i < n; i++) {
        
                // 获取被除数和除数的节点名称
                String dividend = equations.get(i).get(0);
                String divisor = equations.get(i).get(1);
    
                // 如果map中不包含某个节点的字符串，那就添加一个键值对
                if(!map.containsKey(dividend)) {
        
                    map.put(dividend, new ArrayList<>());
                }
                if(!map.containsKey(divisor)) {
        
                    map.put(divisor, new ArrayList<>());
                }
                // 除数和被除数都要放进去，这是一个有向图
                // 除数和被除数的倍数值是倒数关系的
                map.get(dividend).add(new Node(divisor, values[i]));
                map.get(divisor).add(new Node(dividend, 1 / values[i]));
            }
    
            int cnt = 0;
            // 遍历问题
            for(List<String> q : queries) {
        
                // 深搜，初始倍数就是1
                res[cnt] = dfs(q.get(0), q.get(1), 1.0, new HashSet<>());
                cnt ++;
            }
            
            return res;
        } 
    
        // 深搜其中参数cur表示当前节点，dest表示目标节点，knerl表示之前计算的倍数，set保存已经走过的节点
        private double dfs(String cur, String dest, double knerl, Set<String> set) {
        
            // 如果map不包含当前的节点或者已经走过当前节点了，说明这条路不会产生答案
            if(!map.containsKey(cur) || set.contains(cur)) {
        
                return -1.0;
            }
            // 走到了终点，那就返回已经计算了的倍数
            if(cur.equals(dest)) return knerl;
            // 集合中添加当前走过的节点，防止绕圈
            set.add(cur);
    
            // 遍历当前节点的邻接节点
            for(Node node : map.get(cur)) {
        
                // 继续深搜，倍数需要乘上下个一个节点的倍数
                double temp = dfs(node.id, dest, knerl * node.num, set);
                // 如果搜到了答案，就直接返回答案
                if(temp != -1.0) return temp;
            }
    
            // 没有搜到答案，返回-1
            return -1.0;
        }
    }

[Link 1]: https://leetcode-cn.com/problems/evaluate-division/solution/java-wu-xiang-tu-dfs-qing-xi-jie-by-lipp-po5k/