leetcode 日经贴，python code -Copy List with Random Pointer

最新推荐文章于 2019-08-07 21:18:00 发布

原创最新推荐文章于 2019-08-07 21:18:00 发布 · 459 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

python

47 篇文章

订阅专栏

本文介绍了一种复制包含随机指针的单链表的方法，通过插入、重置随机指针和拆分链表三个步骤实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Copy List with Random Pointer

# Definition for singly-linked list with a random pointer.
# class RandomListNode:
#     def __init__(self, x):
#         self.label = x
#         self.next = None
#         self.random = None

class Solution:
    # @param head, a RandomListNode
    # @return a RandomListNode
    def copyRandomList(self, head):
        if head == None: return None
        #insert
        current = head
        while current != None:
            next = current.next
            current.next = RandomListNode(current.label)
            current.next.next = next
            current = next
        #reset random pointer
        current = head
        while current != None:
            next = current.next
            if current.random != None:
                next.random = current.random.next
            current = next.next
        #split
        nhead = head.next
        current, ncurrent = head, nhead
        while current != None:
            current.next = ncurrent.next
            ncurrent.next = ncurrent.next.next if ncurrent.next != None else None
            current = current.next
            ncurrent = ncurrent.next
        return nhead