题目:输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),将此链表进行复制。
复杂链表的结构体定义如下:
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
对于正常链表的复制,很简单只要将链表遍历一遍重新生成一遍就好,那么这个所谓的复杂链表不过是多了一个随机指针(random),其实,我们如果使用暴力的方式,一点一点找也不是不可以的,可能时间复杂度会有点问题,那么我们就来介绍一一种讨巧的思路(这个思路是我看到一位大哥的博客上写的,由于当时没收藏,所以就不给出链接,勿怪)
如果我们先不管随机指针(random),那么原链表应该如图:
那么,有人会问下面那些差不多的单结点干什么的呢?其实,这就是讨巧的地方,如果我们再遍历原结点时,将每个结点插入相同的结点,当然,加入结点的随机指针(random)先设置为NULL,结果如下:
这样就形成了一个新的链表,可能有人会问,这么做什么目的呢?我们看一下,举个例子,假设原链表1结点的random指针指向原链表3结点,那么新链表的1结点的random指针应该指向新的3结点,那么,这个新3结点位置怎么表示呢?很明显,在长链表中就是,原来3结点的next
首先形成新的长链表的代码如下(只需要一个while循环即可):
RandomListNode *p = pHead;
//在每项后面插入一项
while(p)
{
RandomListNode *q = new RandomListNode(p->label);
q->next = p->next;
q->random = NULL;// 新生成结点的random设置为NULL
p->next = q;
p = q->next;
}
接下来我们要做的是给每个新的结点的random赋值,代码如下:
//将添加项的随机指针写入
p = pHead;
while(p)
{
q=p->next;
if(p->random)
q->random=p->random->next;//就是这步对random进行赋值,也就是上面一段比较拗口的话
p=q->next;
}
p = pHead;
然后呢?只需要将一个长链表进行拆分就可以了,拆成两个链表就行了,接下来给出整体代码:
RandomListNode* Clone(RandomListNode* pHead)
{
if(pHead==NULL)
return NULL;
RandomListNode *p = pHead,*q=pHead,*cloneHead;
//在每项后面插入一项
while(p)
{
RandomListNode *q = new RandomListNode(p->label);
q->next = p->next;
q->random = NULL;
p->next = q;
p = q->next;
}
//将添加项的随机指针写入
p = pHead;
while(p)
{
q=p->next;
if(p->random)
q->random=p->random->next;
p=q->next;
}
p = pHead;
//将一个链表拆分
if(p)
{
cloneHead = p->next;
p->next = cloneHead->next;
p = cloneHead->next;
q = cloneHead;
}
RandomListNode *temp=p->next;
while(p)
{
temp = p->next;
p->next = temp->next;
p = p->next;
q->next = temp;
q= temp;
}
return cloneHead;
}