399. Evaluate Division

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]
        """
        #Floyed-Warshall, find all shortest path in a graph
        #dynamic programming 
        #default d[i][j]=inf, if d[i][k]+d[k][j]<d[i][j], replace d[i][j] with smaller value
        #loop till all d[i][j] is smallest possible, or in the case of this problem, when all reachable point has been given value, none reachable points are left blank. 
        d=collections.defaultdict(dict)
        for (numerator,denominator),value in zip(equations,values): 
            d[numerator][numerator]=d[denominator][denominator]=1.0
            d[numerator][denominator]=value
            d[denominator][numerator]=1.0/value
        #find all possible permutations of all the notes in the graph
        for k,i,j in itertools.permutations(d,3):
            if k in d[i] and j in d[k]:
                d[i][j]=d[i][k]*d[k][j]
        return [d[numerator].get(denominator,-1) for (numerator,denominator) in queries]
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容