一、题目描述:86. 分隔链表(中等)
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
1、插入排序的方式,left指针表示链表中节点值大于x的前一个节点。right指针和r_next指针遍历链表,发现r_next的节点值比x小则插入到left指针后面,并且left指针往前挪一步。right指针删除r_next节点(r_next节点往下挪一步。注意边界条件如果初始值left(等于head头节点)节点值大于x,则将r_next节点放在left指针前面,并且left指针指向此时的r_next节点。
三、代码
/**
* 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 * left = head;
if(left == NULL)
return head;
while(left->next != NULL && left->val < x && left->next->val < x)
{
left = left->next;
}
ListNode * right = left;
if(right == NULL)
return head;
ListNode * r_next = right->next;
while(r_next != NULL)
{
if(r_next->val < x)
{
//cout<<"next_va="<<r_next->val<<endl;
if(left->val >= x)
{
head = r_next;
right->next = r_next->next;
r_next->next = left;
r_next = right->next;
left = head;
}
else
{
ListNode * tmp = left->next;
left->next = r_next;
right->next = r_next->next;
r_next->next = tmp;
r_next = right->next;
left = left->next;
}
}
else
{
right = r_next;
r_next = r_next->next;
}
}
return head;
}
};