题目
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
示例 1:
输入:head = [1,4,3,2,5,2], x = 3 输出:[1,2,2,4,3,5]
示例 2:
输入:head = [2,1], x = 2 输出:[1,2]
解题
方法一:
class Solution { public: ListNode* partition(ListNode* head, int x) { ListNode* l1=new ListNode(-1); ListNode* l2=new ListNode(-2); ListNode* cur1=l1; ListNode* cur2=l2; ListNode* cur=head; while(cur){ if(cur->val<x){ ListNode* tmp=cur->next; cur1->next=cur; cur->next=nullptr; cur=tmp; cur1=cur1->next; }else{ ListNode* tmp=cur->next; cur2->next=cur; cur->next=nullptr; cur=tmp; cur2=cur2->next; } } cur1->next=l2->next; return l1->next; } };