Partition List

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.

C++:
01 /**
02  * Definition for singly-linked list.
03  * struct ListNode {
04  *     int val;
05  *     ListNode *next;
06  *     ListNode(int x) : val(x), next(NULL) {}
07  * };
08  */
09 class Solution {
10 public:
11     ListNode *partition(ListNode *head, int x) {
12         // Start typing your C/C++ solution below
13         // DO NOT write int main() function
14         ListNode* smallHead=NULL;
15         ListNode* smallEnd=NULL;
16         ListNode* bigHead=NULL;
17         ListNode* bigEnd=NULL;
18         for(ListNode* i=head;i!=NULL;i=i->next)
19         {
20              if(i->val<x)
21              {
22                  if(smallHead==NULL)
23                  {
24                      smallHead=smallEnd=new ListNode(i->val);
25                  }
26                  else
27                  {
28                      ListNode* item=new ListNode(i->val);
29                      smallEnd->next=item;
30                      smallEnd=item;
31                  }
32              }
33              else
34              {
35                  if(bigHead==NULL)
36                  {
37                      bigHead=bigEnd=new ListNode(i->val);
38                  }
39                  else
40                  {
41                      ListNode* item=new ListNode(i->val);
42                      bigEnd->next=item;
43                      bigEnd=item;
44                  }
45              }
46         }
47         if(smallHead!=NULL)
48         {
49             smallEnd->next=bigHead;
50             return smallHead;
51         }
52         return bigHead;
53     }
54 };
Advertisements
This entry was posted in Array and linked list. Bookmark the permalink.

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s