题目:
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
解题思路:
注意:对于这样的复杂链表,也许刚开始你会认为只要直接进行复制就行,第一次复制节点和next指针,第二次进行双重循环,找到random指针所指的节点并赋值给当前节点的random指针。但这样会有一个问题,当链表中存在重复的元素就会出问题。
所以,最好的解决办法是不使用辅助空间的情况下实现O(N)的时间效率。这道提出的比较巧妙,解题也有一定的技巧性。
1)原始链表的每个结点N创建对应的N',然后将N‘通过next接到N的后面
2)设置复制出来的结点的random指针。N'是N->next指向的结点,那么同样N'->random也是结点N->next->random指向的结点。
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 pHead;
//1、插入新链表
RandomListNode* now = pHead;
RandomListNode* nextNoede;
//RandomListNode* copy;
while(now != NULL){
nextNoede = now->next; //存储当前节点(原链表)的下一个节点
RandomListNode *copy = new RandomListNode(now->label);
now->next = copy;
copy->next = nextNoede;
now = nextNoede;
}
//2、更新新链表的random指针
now = pHead;
while(now != NULL){
if(now->random)
now->next->random = now->random->next;//存储当前节点(原链表)的下一个节点
now = now->next->next;
}
//3、拆分链表
now = pHead;
RandomListNode* mHead = pHead->next;
while(now->next != NULL){
nextNoede = now->next;
now->next = nextNoede->next;
now = nextNoede;
}
return mHead;
}
};