JZ25 复杂链表的复制

复杂链表的复制

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

思路：1)空间复杂度O(1)，时间复杂度O(N). 将新节点先插入至对应原节点的后面，最后将新节点一一拆分出来。

           1、复制每个节点，如：复制节点A得到A1，将A1插入节点A后面

           2、遍历链表，A1->random = A->random->next;

           3、将链表拆分成原链表和复制后的链表

 

2)空间复杂度O(N)，时间复杂度O(N).就是使用一个hashmap存储旧节点和对应的新节点，便于后面找随机节点。

 

func Clone( head *RandomListNode ) *RandomListNode {
    //write your code here
    if head == nil {
        return nil
    }

    // 拷贝每个node
    cur := head
    for cur != nil {
        copyNode := &RandomListNode{Label: cur.Label, Next: cur.Next}
        cur.Next = copyNode
        cur = copyNode.Next
    }

    // 拷贝random
    cur = head
    for cur != nil {
        if cur.Random != nil {
            cur.Next.Random = cur.Random.Next
        }

        if cur.Next != nil {
            cur = cur.Next.Next
        }
    }


    // 断开新老链表
    cur = head
    copyHead := head.Next
    for cur != nil {
        // 1-> 1'->2->2'
        copyNode := cur.Next
        cur.Next = copyNode.Next
        if copyNode.Next != nil {
            copyNode.Next = copyNode.Next.Next
        }
        cur = cur.Next
    }


    return copyHead  
}