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;
}
};
欢迎大家关注我的个人公众号,同样的也是和该博客账号一样,专注分享技术问题,我们一起学习进步