【Graph】399. Evaluate Division（Medium）

#week3#

#from leetcode#

Description

---

 

Equations are given in the format A / B = k, where A and B are variables represented as strings, and k is a real number (floating point number). Given some queries, return the answers. If the answer does not exist, return -1.0.

Example:
Given a / b = 2.0, b / c = 3.0. 
queries are: a / c = ?, b / a = ?, a / e = ?, a / a = ?, x / x = ? . 
return [6.0, 0.5, -1.0, 1.0, -1.0 ].

The input is: vector<pair<string, string>> equations, vector<double>& values, vector<pair<string, string>> queries , where equations.size() == values.size(), and the values are positive. This represents the equations. Return vector<double>.

According to the example above:

equations = [ ["a", "b"], ["b", "c"] ],
values = [2.0, 3.0],
queries = [ ["a", "c"], ["b", "a"], ["a", "e"], ["a", "a"], ["x", "x"] ]. 

 

The input is always valid. You may assume that evaluating the queries will result in no division by zero and there is no contradiction.

 

Analysis

---

 

题目的意思是给我们一些除法等式，让我们通过这些除法等式来求解其他的一些除法式子，输入有等式集，对应等式的值还有查询的等式集。

那么我们应该如何思考这个问题怎么去解决呢？

答案等式只能够在已知的等式推导出来，我们可能直接通过一个等式得到答案，可能是通过几个等式连在一起除下去得到答案；因此我们可以把这个问题想成一个无向图的问题，每一个变量是图中的一个节点，而每一个等式是图中的路径，路径上的权值则是等式的值，为何要想成无向图而不是有向图呢？因为我们式子可以双向推导，另外实际上我们的无向图路径权值不能够只存一个，需要存v->u的权值，还需要有u->v的权值。

想成图之后，我们重新来思考这个问题，求等式的解，那么就是求图中的一个节点是否能够走到另一个节点，如果可以，并求出它们之间的路径的乘积，所以我们这个问题思路就变得很清晰了，那么我们再考虑需要一些什么集合，边集、点集、还有查询集，我们就将最初的equations作为边集，然后再加上相反的边即可，点集在遍历时构建，然后再写一下算法的大概思路如下：

①遍历equations，加入返回边和权重（权重用values数组存储好即可，下标与equations下标一一对应即可），并且将点加入点集；

②遍历queries：

　　对于每个query:

　　　　两点是否在点集中-->no：result=-1，continue继续下一个query；

　　　　-->yes

　　　　　　两个点是否相同-->yes：result=1，continue继续；

　　　　　　-->no

　　　　　　　　从开始点开始做BFS找去终点的路径，如果能够找到则返回权值，如果不能够则result=-1；

于是就可以愉快地编程了~！其实思路是在不断修补完善中的，最开始有的是一个大致框架，然后写的时候发现一些情况漏掉了，再补上去，才把这个算法补成正确完备的算法。

 

Code

---

#include<vector>
#include<string>
#include<queue>
#include<iostream>
using namespace std;
class Solution {
private: 
    static bool exits(vector<string> vec, string value) {
        vector<string>::iterator i;
        for (i = vec.begin(); i != vec.end(); i++) {
            if (value == (*i)) return true;
        }
        return false;
    }
public:
    vector<double> calcEquation(vector<pair<string, string>> equations, vector<double>& values, vector<pair<string, string>> queries) {
        vector<string> points;
        int size = equations.size();
        for (int i = 0; i < size; i++) {
            string firstOne = equations[i].first;
            string secondOne = equations[i].second;
            //cout << (exits(points, firstOne)) << exits(points, secondOne);
            if (!exits(points, firstOne)) points.push_back(firstOne);
            if (!exits(points, secondOne)) points.push_back(secondOne);
            equations.push_back(make_pair(secondOne, firstOne));
            values.push_back(1 / values[i]);
        }

        vector<double> result;
        for (int i = 0; i < queries.size(); i++) {
            string begin = queries[i].first;
            string end = queries[i].second;
        
            if (!exits(points, begin) || !exits(points, end)) {
                result.push_back(-1);
                continue;
            }

            if (begin == end) {
                result.push_back(1);
                continue;
            }

            queue<string> que;
            queue<double> weight;
            vector<string> gone;
            que.push(begin);
            weight.push(1);
            gone.push_back(begin);
            double temp = -1;
            while (!que.empty()) {
                string point = que.front();
                double value = weight.front();
                que.pop();
                weight.pop();
                for (int i = 0; i < equations.size(); i++) {
                    if (equations[i].first == point) {
                        if (equations[i].second == end) {
                            temp = value*values[i];
                            break;
                        }
                        if (!exits(gone, equations[i].second)) {
                            que.push(equations[i].second);
                            weight.push(value*values[i]);
                            gone.push_back(equations[i].second);
                        }
                    }
                }
                if (temp != -1) {
                    break;
                }
            }
            result.push_back(temp);
        }
        return result;
    }
};

Run Time：0ms（leetcode test）