题目描述:
给出一个链表和一个值 ,以 为参照将链表划分成两部分,使所有小于 的节点都位于大于或等于 的节点之前。
两个部分之内的节点之间要保持的原始相对顺序。
例如:
给出 1->4->3->2->5->2和x=3
返回 1→2→2→4→3→5
示例1
输入
{1,4,3,2,5,2},3
返回值
{1,2,2,4,3,5}
分析:
新建两个节点head1与head2,分别为指向两个链表的头结点。
把节点值小于x的节点链接到链表1上,节点值大等于x的节点链接到链表2上。最后把两个链表相连即可
class Solution {
public:
/**
*
* @param head ListNode类
* @param x int整型
* @return ListNode类
/
ListNode partition(ListNode* head, int x) {
// write code here
if(!head) return head;
ListNodehead1=new ListNode(-1);
ListNodehead2=new ListNode(-1);
ListNode* te=head1;
ListNode* tem=head2;
while(head)
{
if(head->val<x)
{
head1->next=head;
head1=head1->next;
}
else
{
head2->next=head;
head2=head2->next;
}
head=head->next;
}
head1->next=tem->next;
head2->next=NULL;
return te->next;
}
};