leetcode 86:Partition List

最新推荐文章于 2022-10-07 19:26:43 发布

原创最新推荐文章于 2022-10-07 19:26:43 发布 · 410 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一种链表分隔算法，该算法可以将链表中小于特定值x的节点放置在大于等于x的节点之前，同时保持原有的相对顺序不变。通过使用两个指针进行操作，确保了时间复杂度为O(n)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

思路：

用两个指针来做，前面一个指针指向的结点，其前面结点对应元素均小于x，后面的指针搜索下一个结点，如果小于x就插入到前一个指针对应结点的后面，否则就跳过。

难点：当头结点改变时的情况。

时间复杂度：O(n)

实现如下：

class Solution {
public:
	ListNode* partition(ListNode* head, int x) {
		ListNode *p = head,*q=head;
		while (p!=NULL && p->val >= x)
		{
			q = p;
			p = p->next;
		}
		if (p == NULL) return head;
		if (p != head)
		{
			q->next = p->next;
			p->next = head;
			head = p;
		}
		if (p->next == NULL) return head;
		while (q != NULL && q->val < x)
		{
			p = q;
			q = q->next;
		}
		if (p == NULL || p->next == NULL) return head;
		while (q->next!=NULL)
		{
			if (q->next->val >= x)
			{
				q = q->next;
				continue;
			}
			ListNode *temp = q->next;
			q->next = q->next->next;
			temp->next = p->next;
			p->next = temp;
			p = p->next;
		}
		return head;
	}
};