栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

剑指 Offer II 111. 计算除法 (先用floyd算法预处理最短路径)

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

剑指 Offer II 111. 计算除法 (先用floyd算法预处理最短路径)

 

直接用floyd处理最短路径,然后直接输出最短路径的值即可。其中使用C++ unordered_map存图技巧。

class Solution {
public:
    vector calcEquation(vector>& equations, vector& values, vector>& queries) {
        unordered_set vertex;
        unordered_map> distance;
        int n = equations.size();
        for (int i = 0; i < n; i++) {
            distance[equations[i][0]][equations[i][1]] = values[i];
            distance[equations[i][1]][equations[i][0]] = 1.0 / values[i]; 
            vertex.insert(equations[i][0]);
            vertex.insert(equations[i][1]);
        }

        for (auto k : vertex) {
            for (auto i : vertex) {
                for (auto j : vertex) {
                    if (distance[i][k] && distance[k][j]) {
                        distance[i][j] = distance[i][k] * distance[k][j];
                    }
                }
            }
        }

        vector res;
        for (auto query : queries) {
            if (distance[query[0]][query[1]]) {
                res.push_back(distance[query[0]][query[1]]);
            } else {
                res.push_back(-1.0);
            }
        }
        return res;
    }
};

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/433494.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号