左神算法基础class3—题目13复制含有随机指针节点的链表

1.题目:复制含有随机指针节点的链表

【题目】 一种特殊的链表节点类描述如下:
public class Node
{
public int value;
public Node next;
public Node rand;
public Node(int data) { this.value = data; }
}
Node类中的value是节点值,next指针和正常单链表中next指针的意义一 样,都指向下一个节点,rand指针是Node类中新增的指针,这个指针可 能指向链表中的任意一个节点,也可能指向null。 给定一个由Node节点类型组成的无环单链表的头节点head,请实现一个 函数完成这个链表中所有结构的复制,并返回复制的新链表的头节点。
进阶:不使用额外的数据结构,只用有限几个变量,且在时间复杂度为 O(N)内完成原问题要实现的函数。

在这里插入图片描述

2.分析

题目给定的数据结构如上图,链表有rand、next两个指针,next指针1->2->3。1的rand指针指向3,2指向1,3的rand指向null,要求复制一份整个链表。
方法一:使用哈希表(哈希表不太熟,以后再补)
方法二:再原来链表基础上修改
思路:单独看1、2、3这三个节点如下图所示,考虑在每个节点后面加入三个新的节点1、2、3连成一个大链表,再分别复制这三个节点,最后分离复制的节点形成一个新链表达到题目要求

在这里插入图片描述
在这里插入图片描述

3.核心代码

(1)生成题设链表

typedef struct ListNode
{
	int val;
	ListNode *next;
	ListNode *rand;
}*List;

①生成头节点

List MakeEmpty()
{
	List head = (List)malloc(sizeof(ListNode));
	head->next = NULL;
	return head;
}

②尾插法插入数据

void insert_next(List head,int num)
{
	//p是新加入的节点
	List p =  (List)malloc(sizeof(ListNode));
	p->val = num;
	p->next = NULL;
	p->rand = NULL;
	
	List r = head;
	while(r->next != NULL)//r指向末尾
	{
		r = r->next;
	}
	r->next = p;//p加入链表
}

③加入rand指针
直接在主函数中添加1->3,2->1,3->NULL

	List cur = head->next;
	cur->rand = cur->next->next;
	cur->next->rand = cur;
	cur->next->next->rand = NULL;

(2)加入新节点

由于加入新节点后,原来链表会被打断,所以需要提前记录下一个节点的位置nex,本步骤相当于在当前位置插入节点。

//加入链
	head = head->next;
	List cur = head;//cur当前节点
	List nex = NULL;
	while(cur != NULL)
	{
		nex = cur->next;
		List p =  (List)malloc(sizeof(ListNode));
		cur->next = p;
		p->next = nex;
		cur = nex;
	}

(3)复制新节点的值及rand指针

例如1->3,那么1*->3*,使用两个指针分别指向当前节点和新加入的节点1、1*。1应该指向1->rand(也就是3)->next(即3),相当于新节点的rand指向旧节点的rand的next。当然,3的rand指向NULL,就没有next指针了,需要加个判断不然会报错。
在这里插入图片描述

	//copy
	cur = head;
	List cur_copy = NULL;//新节点
	while(cur != NULL )
	{
		cur_copy = cur->next;
		cur_copy->rand = cur->rand == NULL ? NULL : cur->rand->next;//拷贝指针
		cur_copy->val = cur->val;//拷贝值
		cur = cur_copy->next;
	}

(4)分离拷贝的部分

直接把新节点连起来,也是通过两个指针cur和cur_copy

//分离拷贝的部分
	List res = head->next;//返回res新节点位置
	cur = head;
	while(cur != NULL)
	{
		nex = cur->next->next;//记录下一个旧节点
		cur_copy = cur->next;
		cur_copy->next = nex == NULL ? NULL : cur_copy->next->next;
		cur = nex;//cur相当于只在旧节点1,2,3之间寻找
	}

4.完整代码

#include<iostream>
using namespace std;

typedef struct ListNode
{
	int val;
	ListNode *next;
	ListNode *rand;
}*List;

List copy_list(List head)
{
	//加入链
	head = head->next;
	List cur = head;//当前节点
	List nex = NULL;
	while(cur != NULL)
	{
		nex = cur->next;
		List p =  (List)malloc(sizeof(ListNode));
		cur->next = p;
		p->next = nex;
		cur = nex;
	}
	//copy
	cur = head;
	List cur_copy = NULL;
	while(cur != NULL )
	{
		cur_copy = cur->next;
		cur_copy->rand = cur->rand == NULL ? NULL : cur->rand->next;
		cur_copy->val = cur->val;
		cur = cur_copy->next;
	}


	//分离拷贝的部分
	List res = head->next;
	cur = head;
	while(cur != NULL)
	{
		nex = cur->next->next;//记录下一个旧节点
		cur_copy = cur->next;
		cur_copy->next = nex == NULL ? NULL : cur_copy->next->next;
		cur = nex;//cur相当于只在旧节点1,2,3之间寻找
	}
	return res;
}

void insert_next(List head,int num)
{
	List p =  (List)malloc(sizeof(ListNode));
	p->val = num;
	p->next = NULL;
	p->rand = NULL;
	List r = head;
	while(r->next != NULL)
	{
		r = r->next;
	}
	r->next = p;
}

List MakeEmpty()
{
	List head = (List)malloc(sizeof(ListNode));
	head->next = NULL;
	return head;
}


int main()
{
	List head = MakeEmpty();
	insert_next(head,5);
	insert_next(head,7);
	insert_next(head,9);
	List cur = head->next;
	cur->rand = cur->next->next;
	cur->next->rand = cur;
	cur->next->next->rand = NULL;

	List res = copy_list(head);

	system("pause");
	return 0;
}
发布了51 篇原创文章 · 获赞 1 · 访问量 1378

猜你喜欢

转载自blog.csdn.net/shi_xiao_xuan/article/details/103763248
今日推荐