Leetcode 399. Evaluate Division

最新推荐文章于 2022-05-08 20:52:48 发布

原创最新推荐文章于 2022-05-08 20:52:48 发布 · 392 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

157 篇文章

订阅专栏

本文深入解析了LeetCode上的题目Evaluate Division（399题），通过构建图模型并使用深度优先搜索算法解决变量之间的比例计算问题。文章详细介绍了如何用Java实现这一算法，包括设置数据结构来存储比例关系以及递归地查找从起点到终点的比例路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Leetcode 399. Evaluate Division

"a"-(2)->"b"

"b"-(1/2)-->"a"-(3)->"c"

深度优先搜索：

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;

public class Solution {
    public double[] calcEquation(String[][] equations, double[] values, String[][] queries) {
    	HashMap<String,ArrayList<String>> pairs = new HashMap<String,ArrayList<String>>();
    	HashMap<String,ArrayList<Double>> pairValues = new HashMap<String,ArrayList<Double>>();
    	double[] results = new double[queries.length];
    	for(int i=0;i<equations.length;i++){
    		String[] equation = equations[i];
    		if(!pairs.containsKey(equation[0])){
    			pairs.put(equation[0], new ArrayList<String>());
    			pairValues.put(equation[0], new ArrayList<Double>());
    		}
    		if(!pairs.containsKey(equation[1])){
    			pairs.put(equation[1], new ArrayList<String>());
    			pairValues.put(equation[1], new ArrayList<Double>());
    		}
    		pairs.get(equation[0]).add(equation[1]);
    		pairValues.get(equation[0]).add(values[i]);
    		pairs.get(equation[1]).add(equation[0]);
    		pairValues.get(equation[1]).add(1/values[i]);
    	}
    	for(int i=0;i<queries.length;i++){
    		String start = queries[i][0],end = queries[i][1];
    		results[i] = dfs(start,end,pairs,pairValues,new ArrayList<String>(),1);
    		if(results[i]==0.0)  results[i] = -1.0;
    	}
    	return results;
    }

	private double dfs(String start, String end, HashMap<String, ArrayList<String>> pairs,
			HashMap<String, ArrayList<Double>> pairValues, ArrayList<String> tmp,double value) {
		// TODO Auto-generated method stub
		if(tmp.contains(start)) return value;
		if(!pairs.containsKey(start)) return 0.0;
		tmp.add(start);
		double res = 0.0;
		for(int i=0;i<pairs.get(start).size();i++){
			String next = pairs.get(start).get(i);
			res = dfs(next,end,pairs,pairValues,tmp,value*pairValues.get(start).get(i));
			if(res!=0.0){
				break;
			}
		}
		tmp.remove(start);
		return res;
	}
}