题目:
题意就是把大于k的放前面,小于k的放后面,但是大于k的那组互相之间相对位置不变,小于k的那组互相之间相对位置也不变。
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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
| 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) { ListNode *virt_node = new ListNode; virt_node->next = head;
ListNode *prev = virt_node, *p, *tmp; while (prev->next != nullptr && prev->next->val < x) { prev = prev->next; }
if (prev->next == nullptr) { return head; } p = prev;
while (p->next != nullptr) { if (p->next->val < x) { tmp = p->next; p->next = tmp->next; tmp->next = prev->next; prev->next = tmp; prev = tmp; } else { p = p->next; } } head = virt_node->next; delete virt_node; return head; } };
|