题目介绍
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例
输入:head = 1 -> 4 -> 3 -> 2 -> 5 -> 2 , x = 3
输出: 1 -> 2 -> 2 -> 4 -> 3 ->5
思路:
将整个链表分为两个链表,一部分是小于x的,一部分是大于等于x的,最后再拼接到一起
注意:1>不是要求升序排列,而只是把小于x的点按原来的排序移到前面,
2>不是真的创建了一个新链表,开辟了新的内存,而是用新的头指针指向原链表中大于等于x的部分,
并将它们依次串起来,而小于x的部分还在原链表中,而且是连续的
3>其余注意点都在代码中进行了注释
代码(这不是唯一的方法,但是是在Leetcode上通过了所有测试用例的)
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* partition(struct ListNode* head, int x) {
if(head == NULL || head->next == NULL)//如果链表是空的或者只有一个结点,直接返回
return head;
struct ListNode* head1 = NULL;//新链表的头指针
struct ListNode* cur = head; //用来遍历原链表
struct ListNode* cur1 = NULL; //指向新链表,用于往新链表中插入新结点,因为不能打乱原来的顺序,所以要尾插
struct ListNode* prev = NULL; //用于连接原链表中所有小于x的结点
while(cur != NULL)
{
if(cur->val < x)//小于x的结点就让它保持不动,还在原链表中
{
prev = cur;
if(head->val >= x)//原链表的头结点应该是第一个小于x的结点
head = prev;
}
else if(cur->val >= x && head1 != NULL)//新链表的头指针指向原链表中第一个大于等于x的结点
{
cur1->next = cur;
cur1 = cur1->next;
}
else if(cur->val >= x && head1 == NULL)//新链表不为空,则往新链表中插入当前大于等于x的结点
{
head1 = cur;
cur1 = head1;
}
if(prev != NULL)
prev->next = cur->next;
cur = cur->next;
}
if(prev != NULL)
prev->next = head1;//此时原链表已经全是小于x的结点了,而且prev指向的是原链表的尾部,
//将原链表和新链表拼接起来
if(cur1 != NULL)
cur1->next = NULL; //新链表只能保证head1到cur1的部分是原链表上拆下来的大于等于x的结点,
//但是cur1后的结点的结点还是原链表上最后一个大于等于x的结点之后的结点
//这块有点绕,需要多体会一下,我在下面画了张图解释这块
return head;
}