【剑指Offer】复杂链表的复制

题目描述

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

解题思路：

1、遍历链表，复制每个结点，如复制结点A得到A1，将结点A1插到结点A后面；

2、重新遍历链表，复制老结点的随机指针给新结点，如A1.random = A.random.next;

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

/*
struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }
};
*/
class Solution {
public:
    RandomListNode* Clone(RandomListNode* pHead){
      if(pHead==NULL)
       return NULL;
        
      RandomListNode* currentNode = pHead;
      //复制每一个节点 
      while(currentNode){
          RandomListNode* cloneNode = new RandomListNode(currentNode->label);//新建节点
          RandomListNode* nextNode = currentNode->next;//获得下个节点 防止插入节点后链表丢失
          currentNode->next = cloneNode;//插入操作
          cloneNode->next = nextNode;
          currentNode = nextNode;//到下个节点
      }
      currentNode = pHead;
       //遍历链表进行指针复制
      while(currentNode){
         //复制的random和原链表的random一致
         currentNode->next->random = currentNode->random==NULL?NULL:currentNode->random->next;
         currentNode = currentNode->next->next;//到下一个节点
      }
      
      currentNode = pHead;//复制前链表
      RandomListNode* pcloneHead = pHead->next;//复制后链表
      while(currentNode){
          RandomListNode* cloneNode = currentNode->next;
          currentNode->next = cloneNode->next;//每次隔一个节点相连，获得复制前的链表
          cloneNode->next=cloneNode->next==NULL?NULL:cloneNode->next->next;
          currentNode = currentNode->next;
      }
      return pcloneHead;
    }
};