Leetcode: Clone Graph

题目

Clone an undirected graph. Each node in the graph contains a label and a list of its neighbors.

 

思路

1. 这题明说, label 独一无二, 那么就可以使用 hash map 存储元素

2. BFS 搜索, 边搜边向 hash map 添加元素

3. 在设置标记位上 TLE 了 N 次, 一个元素一旦被假如到 hash map, 就说明该元素已经被访问到了并已被假如到 queue 中, 同时环的问题也被克服了. 我在做的时候, 把环的问题拉出来单独处理, 但标记忘记了

4. unordered_map<node*, node*> 这种设置不是第一次见到了, 比设置成 unordered_map<int, node*> 要方便一些

5. map.count 比 map.find 要精练一些

6. 加入 map 时, 可以直接 map[] = xxx, 不用判断是否已有

 

代码

class Solution {
public:
    UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
		unordered_map<UndirectedGraphNode*, UndirectedGraphNode*> record;
		if(node == NULL)
			return node;

		deque<UndirectedGraphNode*> queue;
		queue.push_back(node);

		while(!queue.empty()) {
			UndirectedGraphNode *nextNode = queue.front();
			queue.pop_front();

			if(!record.count(nextNode)) {
				UndirectedGraphNode *newNode = new UndirectedGraphNode(nextNode->label);
				record[nextNode] = newNode;
			}
			for(int i = nextNode->neighbors.size()-1; i >= 0 ; i --) {
				UndirectedGraphNode *childNode = nextNode->neighbors[i];
				if(!record.count(childNode)) {
					UndirectedGraphNode *newNode = new UndirectedGraphNode(childNode->label);
					record[childNode] = newNode;
					queue.push_back(childNode);
				}
				record[nextNode]->neighbors.push_back(record[childNode]);
			}
		}
		return record[node];
	}
};