题目描述
给出一个链表和一个值 ,以 为参照将链表划分成两部分,使所有小于 的节点都位于大于或等于 的节点之前。
两个部分之内的节点之间要保持的原始相对顺序。
例如:
给出 1→4 → 3→2 →5→ 2 和 x=3,
返回 1→2→2→4→3→5.
链表结构一定要分清楚遍历到什么位置了,要得到整个链表的结果需要使用链表头结点,而不是每次遍历的节点。链表经常使用的技巧是建立虚拟头结点,是为了在写算法过程中便于处理头结点,比如头结点可能会被删除这类情况。
基本思路是将小于指定数的节点放到一个链表中,把大于等于指定数的节点放入另一个链表中,两个链表拼接起来就构成了题目要求得到的结果。需要首先初始化两个链表的头结点,无论是否遇到比 x 小或者大的数字链表都需要向后移动。
注意:构成一个链表其末尾节点是null。
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode* small = new ListNode(-1);
ListNode* smallH = small;
ListNode* big = new ListNode(-1);
ListNode* bigH = big;
while(head)
{
if(head->val < x)
{
small->next = head;
small = small->next;
}
else
{
big->next = head;
big = big->next;
}
head = head->next;
}
big->next=nullptr;
small->next = bigH->next;
return smallH->next;
}
};