问题描述:
给定一个单链表和数值x,划分链表使得所有小于x的节点排在大于等于x的节点之前。
你应该保留两部分内链表节点原有的相对顺序。
样例给定链表 1->4->3->2->5->2->null,并且 x=3
返回 1->2->2->4->3->5->null
解题思路:
建立两个链表,一个表示小于指定元素的节点,另一个表示大于指定元素的节点,head从链头依次后移,小于指定元素的放入left中,大于指定元素的放入right中,最后将两个链表连接起来。
代码实现:
/**
* Definition of ListNode
* class ListNode {
* public:
* int val;
* ListNode *next;
* ListNode(int val) {
* this->val = val;
* this->next = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param head: The first node of linked list.
* @param x: an integer
* @return: a ListNode
*/
ListNode *partition(ListNode *head, int x) {
// write your code here
if(head==NULL){
return NULL;
}
ListNode *leftnode=new ListNode(0);
ListNode *rightnode=new ListNode(0);
ListNode *left=leftnode;ListNode *right=rightnode;
while(head!=NULL){
if(head->val<x){
left->next=head;left=head;
}else{
right->next=head;right=head;
}
head=head->next;
}
right->next=NULL;
left->next=rightnode->next;
return leftnode->next;
}
};
解题感悟:
将两个链表连起来时要注意逻辑顺序。