categories: [Blog,Algorithm]


86. 分隔链表

难度中等367
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。

示例 1:
86. 分隔链表--🀄️ - 图1
输入:head = [1,4,3,2,5,2], x = 3
输出:[1,2,2,4,3,5]

示例 2:
输入:head = [2,1], x = 2
输出:[1,2]

  1. public ListNode partition(ListNode head, int x) {
  2. ListNode small = new ListNode(0);
  3. ListNode smallHead = small;
  4. ListNode large = new ListNode(0);
  5. ListNode largeHead = large;
  6. while (head != null) {
  7. if (head.val < x) {
  8. small.next = head;
  9. small = small.next;//小链表指针后移
  10. } else {
  11. large.next = head;
  12. large = large.next;//大链表指针后移
  13. }
  14. head = head.next;//遍历主链表
  15. }
  16. large.next = null;
  17. small.next = largeHead.next;//小的指向大的变表头部
  18. return smallHead.next;
  19. }

解析

只需维护两个链表 small 和 large 即可,small 链表按顺序存储所有小于 x 的节点,
large 链表按顺序存储所有大于等于 x 的节点。遍历完原链表后,我们只要将 small 链表尾节点指向
large 链表的头节点即能完成对链表的分隔。

为了实现上述思路,我们设 smallHead 和 largeHead 分别为两个链表的哑节点,即它们的
next 指针指向链表的头节点,这样做的目的是为了更方便地处理头节点为空的边界条件。同时设
small 和 large 节点指向当前链表的末尾节点。开始时
smallHead=small,largeHead=large。随后,从前往后遍历链表,判断当前链表的节点值是否小于
x,如果小于就将 small 的 next 指针指向该节点,否则将 large 的 next 指针指向该节点。

遍历结束后,我们将 large 的 next 指针置空,这是因为当前节点复用的是原链表的节点,而其 next 指针可能指向一个小于 x 的节点,我们需要切断这个引用。同时将 small 的 next 指针指向 largeHead 的 next 指针指向的节点,即真正意义上的 large 链表的头节点。最后返回 smallHead 的 next 指针即为我们要求的答案。