LeetCode86 分割链表

LeetCode86 分割链表

题目

给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

思路

这个题目,先开始是我的思路错误,虽然我知道这个题目需要设置临时头结点,但是由于审题不清,我最先开始是想设置一个临时的结点tempnode,然后小于x就插入tempnode之前,大于或等于x就插入tempnode之后,但是当我看完相关的视频之后,发现题目的要求并不是这样的,因为题目是要求保持相对位置,而如果像我那样插入就会出现这样的情况:
1->2->2->tempnode->5->3->4
这样的结果就和题目要求的相反,看过正确答案后,不得不承认这个题目非常简单,只需要设置两个头结点,temp_less和temp_more,将小于x的值插入temp_less后面,将大于或者等于x的值插入到temp_more后面,然后将temp_less和temp_more链接起来,并将temp_more的next指针赋值为NULL,最后返回temp_less->next即可。

中间出现时间超过限制的错误,最后发现不断有输出数据跳出来,检查发现,temp_more的指针没有赋值为NULL,故不断循环下去。
另外注意的是链接两个链表时要连接头结点!!!而temp_less和temp_more不在是头结点了。
最终代码如下:

//技巧在于巧用临时头结点
	ListNode less_head(0);
	ListNode more_head(0);
	ListNode* tempnode_less = &less_head;
	ListNode* tempnode_more = &more_head;
	ListNode* result = head;
	//if head is not NULL
	while (head)
	{
		//save head 's next
		ListNode* next = head->next;
 
		if (head->val <x)
		{
			//create less x list node
			tempnode_less->next = head;
			tempnode_less = head;
		}
		else
		{
			//create more x list node
			tempnode_more->next = head;
			tempnode_more = head;
		}
 
		//begin next node
		head = next;
	}
	//connect list node, care of "more_head.next"
	tempnode_less->next = more_head.next;
	tempnode_more->next = NULL;
	return less_head.next;

最终用时:超过一小时

上一篇:哈希表的实现-----Java


下一篇:C语言管理系统