题目:

输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)

分析:

拷贝一个带有random指针的链表。

首先想到的便是,复制原链表的的每一个节点,并用next连接起来,在遍历原链表的时候,将原节点和新复制的节点的映射存进map中<Node,Node’>。然后再同时遍历原链表和新链表,此时新链表的random指针指向的元素便是原链表节点指向的节点在map中的映射,也就是A(random)->B,那么A'(random)->(map[A(random)]),因为在map中存储了原节点和新节点的对应关系,可以在O(1)的时间查找到。

还有一种方法是先遍历原链表,将复制的节点插在原链表节点的后面,也就是A->A’->B->B’->NULL,这样的话,复制的节点和原节点也就有了对应的关系,新复制的节点random指向的元素便是原节点random指向的元素的下一个元素,所有新节点random处理完毕后将这个链表再拆分成两个链表即可。

第一步:复制节点,将新创建的节点N’接到N的后面。

 

第二步:复制random指针

 

 

 

第三步:拆分链表

 

程序:

C++

//use hashmap
class Solution {
public:
    RandomListNode* Clone(RandomListNode* pHead)
    {
        RandomListNode* myHead = new RandomListNode(-1);
        RandomListNode* p = pHead;
        RandomListNode* q = myHead;
        while(p != nullptr){
            RandomListNode* tempNode = new RandomListNode(p->label);
            m[p] = tempNode;
            q->next = tempNode;
            p = p->next;
            q = tempNode;
        }
        p = pHead;
        q = myHead->next;
        while(p != nullptr){
            if(p->random != nullptr)
                q->random = m[p->random];
            p = p->next;
            q = q->next;
        }
        return myHead->next;
    }
private:
    map<RandomListNode*, RandomListNode*> m;
};
class Solution {
public:
    RandomListNode* Clone(RandomListNode* pHead)
    {
        if(pHead == nullptr) return nullptr;
        RandomListNode* p = pHead;
        //在每个节点A后复制一个相同的节点A',A->B->C->NULL, A->A'->B->B'->C->C'->NULL
        while(p != nullptr){
            RandomListNode* tempNode = new RandomListNode(p->label);
            tempNode->next = p->next;
            p->next = tempNode;
            p = tempNode->next;
        }
        //复制random指针,A'random指向的一定是Arandom指向的下一个元素。
        p = pHead;
        while(p != nullptr){
            if(p->random != nullptr)
                p->next->random = p->random->next;
            p = p->next->next;
        }
        //将链表拆分,A->A'->B->B'->C->C'->NULL,A->B->C->NULL and A'->B'->C'->NULL
        p = pHead;
        RandomListNode* myHead = pHead->next;
        RandomListNode* q = myHead;
        while(p != nullptr){
            p->next = q->next;
            //如果处理到最后一组节点时,q不用再做处理,因为其next已经指向nullptr
            if(q->next != nullptr)
                q->next = p->next->next;
            p = p->next;
            q = q->next;
        }
        return myHead;
    }
};

Java

import java.util.HashMap;
public class Solution {
    public RandomListNode Clone(RandomListNode pHead)
    {
        RandomListNode myHead = new RandomListNode(-1);
        RandomListNode p = pHead;
        RandomListNode q = myHead;
        while(p != null){
            RandomListNode tempNode = new RandomListNode(p.label);
            q.next = tempNode;
            map.put(p, tempNode);
            p = p.next;
            q = q.next;
        }
        p = pHead;
        q = myHead.next;
        while(p != null){
            if(p.random != null)
                q.random = map.get(p.random);
            p = p.next;
            q = q.next;
        }
        return myHead.next;
    }
    private HashMap<RandomListNode, RandomListNode> map = new HashMap<>();
}

 

版权声明:本文为silentteller原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/silentteller/p/11931355.html