题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
/* struct RandomListNode { int label; struct RandomListNode *next, *random; RandomListNode(int x) : label(x), next(NULL), random(NULL) { } }; */
方法1:拆分为三步走:
1)克隆上述的结点
2)在上述链表基础上增加扩充结点的随机指向
3)把上述的链表拆分出复制的链表
class Solution { public: //复制原始链表的任一节点N并创建新节点N',再把N'链接到N的后边 void CloneNodes(RandomListNode* pHead) { RandomListNode* pNode=pHead; while(pNode!=NULL) { //两个结点对应 两个链表 RandomListNode* pCloned=new RandomListNode(0); pCloned->label=pNode->label;//赋头结点的值 pCloned->next=pNode->next;//PNode的下一个结点即是克隆结点的下一个结点 pCloned->random=NULL; pNode->next=pCloned;//结点 pNode指向克隆链表的头结点 pNode=pCloned->next;//Pnode结点移位到新的结点(若第一个结点是A,则现在对应B) } } //如果原始链表上的节点N的random指向S,则对应的复制节点N'的random指向S的下一个节点S' void ConnectRandomNodes(RandomListNode* pHead) { RandomListNode* pNode=pHead; while(pNode!=NULL) { RandomListNode* pCloned=pNode->next;//pNode头结点指向的下一个结点即是克隆结点 if(pNode->random!=NULL)//random 为随机指向 pCloned->random=pNode->random->next;//明确PNode 和Pcloned 的随机指针的复制关系 pNode=pCloned->next;//移位指针 } } //把得到的链表拆成两个链表,奇数位置上的结点组成原始链表,偶数位置上的结点组成复制出来的链表 RandomListNode* ReConnectNodes(RandomListNode* pHead) { RandomListNode* pNode=pHead; RandomListNode* pClonedHead=NULL; RandomListNode* pClonedNode=NULL; //初始化 if(pNode!=NULL) { pClonedHead=pClonedNode=pNode->next;//新创建的两个链表是为了存储 复制的链表 pNode->next=pClonedNode->next;//新的结点需要隔一个克隆结点的间隔 pNode=pNode->next;//移位到下一个结点 } //循环 while(pNode!=NULL) { pClonedNode->next=pNode->next; pClonedNode=pClonedNode->next;//移位pClonedNode指针 pNode->next=pClonedNode->next; pNode=pNode->next;//移位pNode指针 } return pClonedHead;//返回复制后的链表 pClonedHead=pClonedNode } //3步骤 RandomListNode* Clone(RandomListNode* pHead) { CloneNodes(pHead); ConnectRandomNodes(pHead); return ReConnectNodes(pHead); } };