forked from larissalages/code_problems
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path86.cpp
More file actions
33 lines (33 loc) · 835 Bytes
/
86.cpp
File metadata and controls
33 lines (33 loc) · 835 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(!head || !head->next) return head;
auto l1=new ListNode();
auto l2=new ListNode();
auto l1head=l1;
auto l2head=l2;
while(head){
if(head->val<x){
l1->next=head;
l1=l1->next;
}else{
l2->next=head;
l2=l2->next;
}
head=head->next;
}
l2->next=NULL;
l1->next=l2head->next;
return l1head->next;
}
};