https://leetcode-cn.com/problems/evaluate-division/
分析
这道题的目的很明确,逻辑也容易理解,人可以很容易地计算出这种方程式的结果,但怎么把这种方程式在程序中表示是一个难题。换言之,正确地表示方程式,就能把这个看似难处理的问题转换成容易解决的问题。
解法一、图遍历
如果给出提示,把这题与图联系起来,能否快速找到解决办法呢?
我们可以很自然地想到,可以用带权图来表示输入的所有方程式:a / b = value
表示两条边:a->b
,权重为value
和b->a
,权重为1.0 / value
。对于查询的问题方程式x / y
,问题即可转换为在这个带权图中查找从x
到y
的一条路径,并计算路径上的权重乘积。
几个编码细节
- 如何选择图的表示方法?邻接表。
节点a
是string
类型,所以得用哈希表来存储节点a
到其所有边的映射;用vector
来存储所有由a
指向的节点,即从a
出去的边的集合 - 如何表示带权重的边?结构体
一条加权边除了要具备所指向的节点,还需要有权重,因此定义一个结构体是很好的选择 - 如何找从
x
到y
的路径?BFS或DFS
老生常谈的话题了,只是需要注意存储权重的乘积以及不要遍历已经遍历过的节点。
代码
class Solution {
public:
struct edge {
string node;
double weight;
edge (string node, double weight): node(node), weight(weight) {
}
};
vector<double> calcEquation(vector<vector<string>>& equations, vector<double>& values, vector<vector<string>>& queries) {
unordered_map<string, vector<edge>> edges;
int n = equations.size();
// 构造无向带权图
for (int i = 0; i < n; ++i) {
string x = equations[i][0], y = equations[i][1];
edges[x].push_back(edge(y