Evaluate Division(C++除法求值)

459 篇文章 1 订阅


(1) bfs

class Solution {
public:
    vector<double> calcEquation(vector<vector<string>>& e, vector<double>& v, vector<vector<string>>& q) {
        unordered_map<string,int> mp;
        int number=0;
        for(int i=0;i<e.size();i++) {
            if(mp.find(e[i][0])==mp.end()) {
                mp[e[i][0]]=number;
                number++;
            }

            if(mp.find(e[i][1])==mp.end()) {
                mp[e[i][1]]=number;
                number++;
            }
        }

        vector<vector<pair<int,double>>> edge(number);
        for(int i=0;i<e.size();i++) {
            edge[mp[e[i][0]]].push_back({mp[e[i][1]],v[i]});
            edge[mp[e[i][1]]].push_back({mp[e[i][0]],1/v[i]});
        }

        vector<double> result;
        for(int i=0;i<q.size();i++) {
            double temp=-1.0;
            if(mp.find(q[i][0])!=mp.end() && mp.find(q[i][1])!=mp.end()) {
                int a=mp[q[i][0]];
                int b=mp[q[i][1]];
                if(a==b) {
                    temp=1.0;
                    result.push_back(temp);
                    continue;
                }
                queue<int> qe;
                vector<double> d(number,-1.0);
                d[a]=1.0;
                qe.push(a);
                while(!qe.empty() && d[b]<0) {
                    int t=qe.front();
                    qe.pop();
                    for(int i=0;i<edge[t].size();i++) {
                        if(d[edge[t][i].first]<0) {
                            d[edge[t][i].first]=edge[t][i].second*d[t];
                            qe.push(edge[t][i].first);
                        }
                    }
                }
                temp=d[b];

            }
            result.push_back(temp);
        }
        return result;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值