【leetcode】Clone Graph（python)

最新推荐文章于 2025-09-13 23:06:41 发布

weixin_33858336

最新推荐文章于 2025-09-13 23:06:41 发布

阅读量53

点赞数

文章标签： python 数据结构与算法

本文介绍如何利用深度优先搜索实现无向图的节点复制，并给出具体实现代码。此外，还提供了使用相似方法解决带有随机指针的链表复制问题的思路与代码示例。

类似于二叉树的三种遍历，我们能够基于遍历的模板做非常多额外的事情，图的两种遍历，深度和广度模板相同也能够做非常多额外的事情，这里举例利用深度优先遍历的模板来进行复制，深度优先中，我们先訪问第一个结点，接着訪问第一个邻接点，再訪问邻节点的邻节点。。。。

class Solution:
    # @param node, a undirected graph node
    # @return a undirected graph node
    def cloneGraph(self, node):
        if None == node: return None
        nodeMap = {}
        return self.cloneNode(node, nodeMap)
        
    def cloneNode(self, node, nodeMap):
        if None == node:
            return None
        #訪问当前点，这里不是简单的print，而是复制，若已经复制，则返回副本
        if nodeMap.has_key(node):
            return nodeMap[node]
        #若没有副本，则复制一份，相同处理其邻接点
        else:
            clone = UndirectedGraphNode(node.label)
            nodeMap[node] = clone
            #訪问其邻居节点
            for neighbor in node.neighbors:
                clone.neighbors.append(self.cloneNode(neighbor, nodeMap))
        return clone

与这个题类似，

Copy List with Random Pointer

这个题目中假设同意使用额外的空间，我们也能够用这样的办法来获得一份拷贝。

class Solution:
    # @param head, a RandomListNode
    # @return a RandomListNode
    def copyRandomList(self, head):
        if None == head: return None
        nodeMap = {}
        return self.copyListNode(head, nodeMap)
        
    def copyListNode(self, node, nodeMap):
        if None == node: return None
        if nodeMap.has_key(node):
            return nodeMap[node]
        else: 
            cpNode = RandomListNode(node.label)
            nodeMap[node] = cpNode
            cpNode.next = self.copyListNode(node.next, nodeMap)
            cpNode.random = self.copyListNode(node.random, nodeMap)
        return cpNode