LeetCode-133.克隆图(相关话题:广度优先)

克隆一张无向图,图中的每个节点包含一个 label (标签)和一个 neighbors (邻接点)列表 。

OJ的无向图序列化:

节点被唯一标记。

我们用 # 作为每个节点的分隔符,用 , 作为节点标签和邻接点的分隔符。

例如,序列化无向图 {0,1,2#1,2#2,2}。

该图总共有三个节点, 被两个分隔符 # 分为三部分。 

  1. 第一个节点的标签为 0,存在从节点 0 到节点 1 和节点 2 的两条边。
  2. 第二个节点的标签为 1,存在从节点 1 到节点 2 的一条边。
  3. 第三个节点的标签为 2,存在从节点 2 到节点 2 (本身) 的一条边,从而形成自环。

我们将图形可视化如下:

   1
  / \
 /   \
0 --- 2
     / \
     \_/

Java代码:

public UndirectedGraphNode cloneGraph(UndirectedGraphNode node) {
    if (null == node)
        return null;

    UndirectedGraphNode head = new UndirectedGraphNode(node.label);
    Queue<UndirectedGraphNode> queue = new java.util.concurrent.LinkedBlockingQueue<>();
    Queue<UndirectedGraphNode> queue1 = new java.util.concurrent.LinkedBlockingQueue<>();
    Map<UndirectedGraphNode, UndirectedGraphNode> map = new HashMap<>();
    queue.add(node);
    queue1.add(head);
    map.put(node, head);
    while (!queue.isEmpty()) {
        UndirectedGraphNode cur = queue.poll();
        UndirectedGraphNode tmp = queue1.isEmpty() ? new UndirectedGraphNode(cur.label) : queue1.poll();
        for (UndirectedGraphNode graphNode: cur.neighbors) {
            UndirectedGraphNode graphNode1 = null == map.get(graphNode) ? new UndirectedGraphNode(graphNode.label) : map.get(graphNode);
            tmp.neighbors.add(graphNode1);
            if (null == map.get(graphNode)) {
                queue.add(graphNode);
                queue1.add(graphNode1);
                map.put(graphNode, graphNode1);
            }
        }
    }

    return head;
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/85212025
今日推荐