leetcode 399. 除法求值

最新推荐文章于 2021-08-19 21:26:50 发布

原创最新推荐文章于 2021-08-19 21:26:50 发布 · 119 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一种使用哈希表解决方程式求解问题的算法，通过构建元素池实现快速读取，适用于处理一系列给定的变量比例关系，如a/b=2.0, b/c=3.0等，能高效求解未知比例。

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

给出方程式 A / B = k, 其中 A 和 B 均为用字符串表示的变量， k 是一个浮点型数字。根据已知方程式求解问题，并返回计算结果。如果结果不存在，则返回 -1.0。

示例 :
给定 a / b = 2.0, b / c = 3.0
问题: a / c = ?, b / a = ?, a / e = ?, a / a = ?, x / x = ?
返回 [6.0, 0.5, -1.0, 1.0, -1.0 ]

输入为: vector<pair<string, string>> equations, vector<double>& values, vector<pair<string, string>> queries(方程式，方程式结果，问题方程式)，其中 equations.size() == values.size()，即方程式的长度与方程式结果长度相等（程式与结果一一对应），并且结果值均为正数。以上为方程式的描述。返回vector<double>类型。

基于上述例子，输入如下：

equations(方程式) = [ ["a", "b"], ["b", "c"] ],
values(方程式结果) = [2.0, 3.0],
queries(问题方程式) = [ ["a", "c"], ["b", "a"], ["a", "e"], ["a", "a"], ["x", "x"] ].
输入总是有效的。你可以假设除法运算中不会出现除数为0的情况，且不存在任何矛盾的结果。

哈希表

class Solution(object):
    def calcEquation(self, equations, values, queries):
        """
        :type equations: List[List[str]]
        :type values: List[float]
        :type queries: List[List[str]]
        :rtype: List[float]
        """
        res = []
        dic_list = [{}]
        length = len(equations)
        for i in range(length):
            num1 = equations[i][0]
            num2 = equations[i][1]
            num1Todic = num2Todic = None
            dic1Index = dic2Index = None
            for j in range(len(dic_list)):
                dic = dic_list[j]
                if num1 in dic:
                    num1Todic = dic
                    dic1Index = j
                if num2 in dic:
                    num2Todic = dic
                    dic2Index = j
            if not num1Todic and not num2Todic:
                new_dic={}
                new_dic[num2] = 1
                new_dic[num1] = values[i]
                dic_list.append(new_dic)
            elif not num1Todic:
                num2Todic[num1] = values[i] * num2Todic[num2]
            elif not num2Todic:
                num1Todic[num2] = num1Todic[num1] / values[i]
            else:
                for key in num1Todic.keys():
                    num1Todic[key] *= (values[i] *  num2Todic[num2] / num1Todic[num1])
                num2Todic.update(num1Todic)
                dic_list.pop(dic1Index)
        for i in range(len(queries)):
            num1 = queries[i][0]
            num2 = queries[i][1]
            dic1Index = dic2Index = None
            for j in range(len(dic_list)):
                if num1 in dic_list[j]:
                    dic1Index = j
                if num2 in dic_list[j]:
                    dic2Index = j
            if dic1Index == dic2Index and dic1Index != None:
                res.append(dic_list[dic1Index][num1]/dic_list[dic1Index][num2])
            else:
                res.append(float(-1))
        return res

用哈希表来构建有相关大小关系的元素池。这样可以做到O(1)的读取速度。当两个相除元素不在同一个池（哈希表）中时，需要对池进行合并。不然只需要依靠现有关系在池中添加元素就可以了。