编程题---分隔链表

题目介绍

给定一个链表和一个特定值 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;
}

这里写图片描述

猜你喜欢

转载自blog.csdn.net/it_xiaoye/article/details/80468473