剑指offer21 --- 复杂链表的复制

复杂链表的复制

题目

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针random指向一个随机节点），请对此链表进行深拷贝，并返回拷贝后的头结点。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

分析

• 解法1:

  deepcopy可以但是copy不行.

  import copy
  def Clone(self,pHead)
  	ret = copy.deepcopy(pHead)
      return ret
  

• 解法2：

  用链表实现。每一次都要搜random的指针，复杂度为$n^2$,很慢。怎么办？ —>

  
  

复杂度O(n)

解

# -*- coding:utf-8 -*-
# class RandomListNode:
#     def __init__(self, x):
#         self.label = x
#         self.next = None
#         self.random = None
class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        if pHead == None:
            return None
        #复制一个一样的node，并且添加到之前链表的每一个node后面
        pTmp=pHead
        while pTmp:
            node = RandomListNode(pTmp.label)
            node.next=pTmp.next
            pTmp.next=node
            pTmp=node.next
        # 实现新建的random的指向
        pTmp=pHead
        while pTmp:
            if pTmp.random:
                pTmp.next.random=pTmp.random.next
            pTmp=pTmp.next.next
            
        # 断开原来的node和新node之间的链接
        pTmp=pHead
        newHead=pHead.next
        pNewTmp=pHead.next
        while pTmp:
            pTmp.next=pTmp.next.next
            if pNewTmp.next:
                pNewTmp.next=pNewTmp.next.next
                pNewTmp=pNewTmp.next
            pTmp=pTmp.next
        return newHead

牛客网该题链接