题目描述
本题要求复制一个无向图,图中每个节点都包含一个标签和它的邻居列表
我们无向图用以下的方法序列化:
• 节点的标签是互不相同的,
• 我们使用“#”作为节点之间的分隔符,使用“,”作为节点标签和节点的节点邻居的分隔符。
例如:现在有一个序列化的无向图{0,1,2#1,2#2,2}.
这个无向图一共有3个节点,因此序列被#分隔成三部分
- 第一个节点的标签是0,节点0和节点1,节点2之间有边
- 第二个节点的标签是1,节点1和节点2之间有边
- 第三个节点的标签是2,节点2和节点2(它自己)之间有边,形成了自环
分析
对图进行广度优先遍历,因为无向图定义的类方法在new一个无向图结点时只能把标签复制下来,它的邻居列表是无法复制的,因此需要一个哈希表对无向图原结点和复制的结点映射,方便用这个映射找到它的复制结点然后修改邻居列表。
代码
/**
* Definition for undirected graph.
* class UndirectedGraphNode {
* int label;
* ArrayList<UndirectedGraphNode> neighbors;
* UndirectedGraphNode(int x) { label = x; neighbors = new ArrayList<UndirectedGraphNode>(); }
* };
*/
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
public class Solution {
public UndirectedGraphNode cloneGraph(UndirectedGraphNode node) {
if(node == null)return null;
//BFS遍历无向图
LinkedList<UndirectedGraphNode> queue = new LinkedList<>();
//记录已经遍历过的结点,分别是原结点和复制节点
HashMap<UndirectedGraphNode,UndirectedGraphNode> map = new HashMap<>();
//因为方法问题只能复制标签不能把它的邻居列表也复制
UndirectedGraphNode newnode = new UndirectedGraphNode(node.label);
map.put(node,newnode);
UndirectedGraphNode copy;
queue.offer(node);
while(!queue.isEmpty()){
UndirectedGraphNode p = queue.poll();
for(UndirectedGraphNode n : p.neighbors){
//如果没有邻居遍历过
if(!map.containsKey(n)){
queue.offer(n);
copy = new UndirectedGraphNode(n.label);
map.put(n,copy);
}
//遍历过的话找到这个复制点
else{
copy = map.get(n);
}
map.get(p).neighbors.add(copy);
}
}
return newnode;
}
}