链表问题---复制含有随机指针节点的链表

【题目】

  一种特殊的链表节点类型描述如下:

class Node:
    def __init__(self, data):
        self.val = data
        self.next = None
        self.rand = None

  Node类中的value是节点值,next指针和正常单链表中next指针的意义一样,都指向下一个节点,rand指针是Node类中新增的指针,这个指针可能指向链表中的任意一个节点,也可能指向None。
  给定一个由Node节点类型组成的无环单链表的头节点head,请实现一个函数完成这个链表中所有结构的复制,并返回复制的新链表的头节点。

  进阶:不使用额外的数据结构,只用有限几个变量,且在时间复杂度为O(N)内完成原问题要实现的函数。

【基本思路】

  原问题。使用一个哈希表,建立旧节点和新节点的对应关系。然后遍历一遍链表,根据原链表和哈希表设置每一个新节点的next和rand指针。

#python3.5
class RandNode:
    def __init__(self, data):
        self.val = data
        self.next = None
        self.rand = None

def copyListWithRand1(head):
    if head == None:
        return None
    map = {}
    cur = head
    while cur != None:
        map[cur] = RandNode(cur.val)
        cur = cur.next
    cur = head
    while cur != None:
        map[cur].next = None if cur.next == None else map[cur.next]
        map[cur].rand = None if cur.rand == None else map[cur.rand]
        cur = cur.next
    return map[head] 

  进阶问题。在原链表中每一个节点的后面插入一个副本节点。遍历一遍链表,在遍历时设置每一个副本节点的rand指针。最后再分离成两个链表即可。

def copyListWithRand2(head):
    if head == None:
        return None
    cur = head
    while cur != None:
        next = cur.next
        cur.next = RandNode(cur.val)
        cur.next.next = next
        cur = next
    cur = head
    while cur != None:
        cur.next.rand = None if cur.rand == None else cur.rand.next
        cur = cur.next.next
    copyHead = head.next
    cur = head
    while cur != None:
        next = cur.next
        cur.next = next.next
        next.next = None if next.next == None else next.next.next
        cur = cur.next
    return copyHead
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值