leetcode 86.分割链表

leetcode 86.分割链表

题目描述

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

解题思路

题目要求不改变每个节点的相对位置，这里采用双指针的方法，使用一个链表指针记录小于x的数，使用另外一个链表指针记录大于等于x的数，最后把两个链表合并。

/**
 * 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) {
        if(!head){       // 如果链表为空直接返回
            return NULL;
        }

        ListNode* cur = head;   // 定义变量用于链表节点的移动
        ListNode* less = new ListNode(-1); // 小于x的节点存放less中
        ListNode* lessPtr = less;          // 用于移动less链表中的节点
        ListNode* more = new ListNode(-1); // 大于等于x的节点存放在more中
        ListNode* morePtr = more;          // 用于移动more链表中的节点 
        while(cur){
            if(cur->val < x){              // 当节点中的值小于x时
                lessPtr->next = cur;       // 在less中添加节点
                lessPtr = lessPtr->next;   // 移动lessPtr指针的位置，始终指向尾部
            }
            else{                          // 当节点中的值大于等于x时
                morePtr->next = cur;       // 在more中添加节点
                morePtr = morePtr->next;   // 移动morePtr指针的位置，始终指向尾部
            }
            cur = cur->next;     
        }
        morePtr->next = NULL;              // 遍历完成more尾部的节点指向空
        lessPtr->next = more->next;        // less尾部节点指向more->next，合并两个链表
        return less->next;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步