Difficulty: 3
Frequency: 3
Problem:
Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal tox.
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
.
Solution:
/**
* 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) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
if (head==NULL)
return NULL;
ListNode * lessTraverse = NULL, * moreTraverse = NULL, * p_traverse = head, * moreHead = NULL;
head = NULL;
while(p_traverse!=NULL)
{
if (p_traverse->val<x)
{
if (lessTraverse==NULL)
{
head = p_traverse;
lessTraverse = p_traverse;
}
else
{
lessTraverse->next = p_traverse;
lessTraverse = lessTraverse->next;
}
p_traverse = p_traverse->next;
}
else
{
if (moreTraverse==NULL)
{
moreHead = p_traverse;
moreTraverse = p_traverse;
}
else
{
moreTraverse->next = p_traverse;
moreTraverse = moreTraverse->next;
}
p_traverse = p_traverse->next;
}
}
if (head!=NULL)
{
lessTraverse->next = moreHead;
if (moreTraverse)
moreTraverse->next = NULL;
return head;
}
else
{
moreTraverse->next = NULL;
return moreHead;
}
}
};
Notes:
This problem has lots of subtleties.