复杂链表的复制

复杂链表的复制

题目描述

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

思路

本体的难点在于特殊指针的复制，解决方法是先将每个原节点复制一遍放在对应原节点的后面，原节点之间的相对位置和新节点之间的相对位置是一致的，这样以来就能很容易的将特殊指针也复制到新节点上了。最后再将原链表拆分成一老一新两个链表。

第一步：复制原节点并依次插入

第二步：根据相对位置复制原节点的特殊指针

第三步：将链表拆分得到深拷贝后的链表

Python 实现

# -*- 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 not pHead:
            return pHead
        p = pHead
        while(p):
            tmp_p = RandomListNode(p.label)
            tmp_p.next = p.next
            p.next = tmp_p
            p = p.next.next
        p = pHead
        while(p):
        # 这里要注意特殊指针指向None的特殊情况
            p.next.random = p.random.next if p.random else None
            p = p.next.next
        p = pHead
        p_tmp = pHead.next
        new_p = pHead.next
        while(p):
            p.next = p_tmp.next
            p_tmp.next = p.next.next if p.next else None
            p = p.next
            p_tmp = p_tmp.next
        return new_p

测试

牛客