Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.
You should preserve the original relative order of the nodes in each of the two partitions.
For example,
Given 1->4->3->2->5->2
and x = 3,
return 1->2->2->4->3->5
.
1、最容易想到的方法就是分别建立两个链表,一个放大于给定值的,一个放小于给定值的,最后合并,但按这种思路实现最后超时了。
2、参考方法二,稍微改了一下思路就能通过:
将小于给定值的部分组成一个链表,剩余部分组成一个链表,不同的地方在于,剩余部分直接在链表上修改,而不是重新创建链表。
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(!head) return head;
ListNode* dummy = new ListNode(-1);
dummy->next = head;
ListNode* newdummy = new ListNode(-1);
ListNode* cur = dummy, *p = newdummy;
while(cur->next)
{
if(cur->next->val < x)
{
p->next = cur->next;
p = p->next;
cur->next = cur->next->next;
p->next = nullptr;
}
else
{
cur = cur->next;
}
}
p->next = dummy->next;
return newdummy->next;
}
};