当前位置: 代码迷 >> 综合 >> leetcode 399. Evaluate Division (medium)
  详细解决方案

leetcode 399. Evaluate Division (medium)

热度:96   发布时间:2024-01-05 00:20:57.0
题目链接

思路:

建立图结构,(节点可以用自定义结构体或类,也可以用容器配合pair)被除数/除数作为节点,之间的商作为节点间的权重值。
转换成图后,题目可以转换为求两点间的权重值。因为不需要寻求最短路径,所以dfs和bfs都可,这里使用dfs。

class Solution
{
    
public:unordered_map<string, vector<pair<string, double>>> mGraph;vector<double> calcEquation(vector<vector<string>> &equations, vector<double> &values, vector<vector<string>> &queries){
    vector<double> res;// 建立图for (int i = 0; i < equations.size(); i++){
    mGraph[equations[i][0]].push_back({
    equations[i][1], values[i]});mGraph[equations[i][1]].push_back({
    equations[i][0], 1 / values[i]});}// 求结果for (int i = 0; i < queries.size(); i++){
    res.push_back(dfs(queries[i][0], queries[i][1]));}return res;}double dfs(const string &src, const string &des){
    if (!mGraph.count(src) || !mGraph.count(des)){
    return -1.0;}if (src == des){
    return 1.0;}queue<pair<string, double>> que;unordered_set<string> visited;que.push({
    src, 1.0});while (!que.empty()){
    pair<string, double> mSrcVal = que.front();que.pop();if (mSrcVal.first == des){
    return mSrcVal.second;}visited.insert(mSrcVal.first);for (auto vec : mGraph[mSrcVal.first]){
    if (!visited.count(vec.first)){
    que.push({
    vec.first, vec.second * mSrcVal.second});}}}return -1;}
};
  相关解决方案