剑指offer 复杂链表的复制

剑指offer题型分类及各题的代码及解题思路

题目描述

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

节点定义如下：

struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }
};

分析：

1、复制每个节点，如：复制节点A得到A1，将A1插入节点A后面

2、遍历链表，A1->random = A->random->next;

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

 

C++实现代码如下：

RandomListNode* Clone(RandomListNode* pHead)
{
	if (NULL == pHead)
		return NULL;
	
	// 1、复制每个节点，如：复制节点A得到A1，将A1插入节点A后面
	RandomListNode* currNode = pHead;
	while (currNode)
	{
		RandomListNode* node = new RandomListNode(currNode->label);    //开辟空间并初始化新结点
		node->next = currNode->next;
		currNode->next = node;
		currNode = node->next;
	}
	
	// 2、遍历链表，A1->random = A->random->next;
	currNode = pHead;
	while (currNode)
	{
		RandomListNode* node = currNode->next;
		if (currNode->random)
		{
			node->random = currNode->random->next;
		}
		currNode = node->next;
	}
	
	// 3、将链表拆分成原链表和复制后的链表
	currNode = pHead;
	RandomListNode* temp;
	RandomListNode* pCloneHead = pHead->next;
	while (currNode->next)
	{
		temp = currNode->next;
		currNode->next = temp->next;
		currNode = temp;
	}
	return pCloneHead;
}