复杂链表的复制

#include <stdlib.h>
#include <stdio.h>

typedef struct RandomListNode
{
  int _val;
  struct RandomListNode* _next;
  struct RandomListNode* _random;
}RandomListNode;

class solution
{
  public:
    RandomListNode* BuyNode(int val)
    {
      RandomListNode* newNode = (RandomListNode*)malloc(sizeof(RandomListNode));
      newNode->_next = NULL;
      newNode->_random = NULL;
      newNode->_val = val;
      return newNode;
    }

    RandomListNode* Clone(RandomListNode* pHead)
    {
      if(NULL == pHead)
        return NULL;
      RandomListNode* pCur = pHead;
	  //复制新节点插入到当前节点后面
      while(pCur)
      {
        RandomListNode* Node = pCur;
        Node->_next = pCur->_next;
        pCur->_next = Node;
        pCur = Node->_next;
      }
	  //设置随机节点
      pCur = pHead;
      while(pCur)
      {
        RandomListNode* Node = pCur->_next;
        if(pCur->_random)
        {
          Node->_random = pCur->_random->_next;
        }
        pCur = Node->_next;
      }
	  //拆分链表
      RandomListNode* pCloneHead = pHead->_next;
      RandomListNode* tmp;
      pCur = pHead;
      while(pCur->_next)
      {
        tmp = pCur->_next;
        pCur->_next = tmp->_next;
        pCur = tmp;
      }
      return pCloneHead;
    }
};
//测试
int main()
{
  solution s;
  RandomListNode* node1 = s.BuyNode(1);
  RandomListNode* node2 = s.BuyNode(2);
  RandomListNode* node3 = s.BuyNode(3);
  
  node1->_next = node2;
  node1->_random = node3;
  node2->_next = node3;
  node2->_random = node2;
  node3->_next = NULL;
  node3->_random = node1;
  RandomListNode* newNode = s.Clone(node1);
  while(newNode)
  {
    printf("%d ",newNode->_val);
    newNode = newNode->_next;
  }
  return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值