【算法笔记】LeetCode_86 分隔链表-编程知识

【算法笔记】LeetCode_86 分隔链表

LeetCode_86 分隔链表

LeetCode_86 分隔链表

题目描述

给你一个链表的头节点 head 和一个特定值 x ，请你对链表进行分隔，使得所有小于 x 的节点都出现在 大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例 1：

输入：head = [1,4,3,2,5,2], x = 3
输出：[1,2,2,4,3,5]

示例 2：

输入：head = [2,1], x = 2
输出：[1,2]

提示：

链表中节点的数目在范围 [0, 200] 内
-100 <= Node.val <= 100
-200 <= x <= 200

思路

这道题算是中等难度里面比较简单的了，但有个小坑困扰了我半个小时，特此记录一下：

题目整体思路很简单，解答的时候可以想象在给苹果分类，比如给你一批苹果，大小不一，然后你面前有两个桶，左边的桶写着（直径大于15cm），右边的桶写着（其他），然后你依次测量苹果直径然后放进两个桶内，至于题目说的“保持原始的次序”，由于链表的特性，这个不是问题。

所以我们初始化两个局部对象ListNode * leftPart， rightPart，然后对原始链表进行遍历，如果当前节点的val小于x就连接到leftPart，否则连接到rightPart，最后，连接leftPart的末尾和rightPart的开头，非常简单。

我说的小坑是：一定不要忘记将新链表的最后一个node指向nullptr，否则整个链表可能产生环，运行不过

代码解析

class Solution {  
public:  ListNode* partition(ListNode* head, int x) {  // 如果链表为空或只有一个节点，则无需分区，直接返回原链表  if (head == nullptr || head->next == nullptr)  {  return head;  }  // 初始化左、右两个链表的虚拟头节点  ListNode * leftPart = new ListNode();  ListNode * leftHead = leftPart;  ListNode * rightPart = new ListNode();  ListNode * rightHead = rightPart;  // 遍历原链表  while(head != nullptr)  {  // 如果当前节点的值小于x，将其加入左链表  if (head->val < x)  {  leftPart->next = head;  leftPart = leftPart->next;  }  // 如果当前节点的值大于等于x，将其加入右链表  else  {  rightPart->next = head;  rightPart = rightPart->next;  }  // 移动到原链表的下一个节点  head = head->next;  }  // 将右链表的末尾置为nullptr，防止成环  rightPart->next = nullptr;  // 将左链表的末尾指向右链表的头节点，形成新的链表  leftPart->next = rightHead->next;  // 返回新链表的头节点（左链表的第二个节点）  return leftHead->next;  }  
};