题目描述:
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
方法1:
主要思路:
(1)使用两个辅助头结点,分别保存链表的小于x和大于等于x 的两部分;
(2)最后将两部分进行合并;
/**
* 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) {
//辅助头结点
ListNode* dummy1=new ListNode(0);
ListNode* dummy2=new ListNode(0);
ListNode* cur_head1=dummy1;
ListNode* cur_head2=dummy2;
//分割链表
while(head){
if(head->val<x){
cur_head1->next=head;
head=head->next;
cur_head1=cur_head1->next;
}
else{
cur_head2->next=head;
head=head->next;
cur_head2=cur_head2->next;
}
}
//合并两个链表
cur_head2->next=NULL;
cur_head1->next=dummy2->next;
head=dummy1->next;
delete dummy1;
delete dummy2;
return head;
}
};