leetcode之Partition List

最新推荐文章于 2025-08-13 22:03:49 发布

原创最新推荐文章于 2025-08-13 22:03:49 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #面试

本文详细阐述了如何通过双指针技巧对链表进行分区，确保所有值小于给定值x的节点位于前半部分，而大于等于x的节点位于后半部分，同时保持原始相对顺序不变。

摘要生成于 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的那一段的结尾

注意，当目前遍历的和尾部相同时候直接全部赋值为下一个即可

/**
* Definition for singly-linked list.
* struct ListNode {
*     int val;
*     ListNode *next;
*     ListNode(int x) : val(x), next(NULL) {}
* };
*/

class Solution {
public:
	ListNode* partition(ListNode* head, int x) {
		ListNode *phead = new ListNode(0);
		phead->next = head;
		ListNode *cur = phead;
		ListNode *small = phead;
		while (cur->next)
		{
			if (cur->next->val < x)
			{
			    if(cur == small)
			    {
			        cur = cur->next;
			        small  = small->next;
			    }
			    else
			    {
				    ListNode *tmp = cur->next;
				    cur->next = cur->next->next;
				    tmp->next = small->next;
				    small->next = tmp;
				    small = small->next;
			    }
			}
			else
				cur = cur->next;
		}
		return phead->next;
	}
};