这道题本身没什么难度。和 100. Copy List with Random Pointer 其实是一回事。
但这道题首次提出了 undirected graph 的串行化的表示方法,却让我想了半天。
说实话,如何将 undirected graph 用 LeetCode 的串行化方式打印出来,比这道题本身难得多。
若有心有成竹的朋友,可以在 #4 里试试。
这道题本身的话,用 DFS 即可解决,模仿 100 题的解答,我们依然用一张 Hash 表来对照着拷贝。
其余就无需多说了,代码也非常短。效率倒是一般,中间水平。
#include <vector>
using std::vector;
#include <unordered_map>
using std::unordered_map;
struct UndirectedGraphNode {
int label;
vector<UndirectedGraphNode *> neighbors;
UndirectedGraphNode(int x) : label(x) {};
};
class Solution {
unordered_map<UndirectedGraphNode*, UndirectedGraphNode*> map;
void dfs(UndirectedGraphNode *node) {
if (map.find(node) != map.end()) return;
map[node] = new UndirectedGraphNode(node->label);
for (UndirectedGraphNode *neighbor : node->neighbors) {
dfs(neighbor);
map[node]->neighbors.push_back(map[neighbor]);
}
}
public:
UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
if (!node) return node;
dfs(node);
return map[node];
}
};