(动画详解)LeetCode面试题 02.04.分割链表

💖💖💖欢迎来到我的博客，我是anmory💖💖💖
又和大家见面了
欢迎来到动画详解LeetCode系列
用通俗易懂的动画的动画使leetcode算法题可视化
先来自我推荐一波
个人网站欢迎访问~~以及捐款~~
推荐阅读
如何低成本搭建个人网站
专栏:动画详解leetcode算法题
C语言知识

题目描述

解题思路

本题可以通过两个链表来解决
一个链表是小链表，用来存储比x小的元素
一个链表是大链表，用来存储大于等于x的元素
最后再将大小链表首尾相接就可以了

动画详解

动画详解分解链表

代码实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */

typedef struct ListNode ListNode;

struct ListNode* partition(struct ListNode* head, int x)
{
    ListNode* lessHead, *lessTail;
    ListNode* greatHead,*greatTail;
    lessHead = lessTail = (ListNode*)malloc(sizeof(ListNode));
    greatHead = greatTail = (ListNode*)malloc(sizeof(ListNode));

    ListNode* pcur = head;
    while(pcur)
    {
        if(pcur->val<x)
        {
            // 插入到小链表中
            lessTail->next = pcur;// 这里的思想可以学习一下
            lessTail = lessTail->next;
        }
        else
        {
            // 插入到大链表中
            greatTail->next = pcur;
            greatTail = greatTail->next;
        }
        pcur = pcur->next;
    }
    // 把大链表的尾部置空，防止出现链表循环
    greatTail->next = NULL;
    // 大小链表首尾相接
    lessTail->next = greatHead->next;
    return lessHead->next;
}