力扣题解：面试题 02.04. 分割链表

最新推荐文章于 2022-11-21 15:21:22 发布

胡矣

最新推荐文章于 2022-11-21 15:21:22 发布

阅读量345

点赞数 1

分类专栏：算法 LeetCode 文章标签： leetcode题解力扣题解链表排序分割链表

本文链接：https://blog.csdn.net/yirentianran/article/details/121671031

版权

算法同时被 2 个专栏收录

142 篇文章 0 订阅

订阅专栏

LeetCode

130 篇文章 0 订阅

订阅专栏

题目

给你一个链表的头节点 head 和一个特定值 x ，请你对链表进行分隔，使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。
你不需要保留每个分区中各节点的初始相对位置。
示例 1：
输入：head = [1,4,3,2,5,2], x = 3
输出：[1,2,2,4,3,5]

解题思路

当前节点curr

curr >= x，继续检查下个节点
curr < x
- curr左边不存在 >= x的节点，继续检查下个节点
- curr左边存在 >= x的节点，将curr节点移动到链表最前面

代码

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
  public ListNode partition(ListNode head, int x) {
    // prev：前一个节点，curr：当前节点
    ListNode dummy = new ListNode(), prev = dummy, curr = head;
    dummy.next = head;
    // 是否需要移动
    boolean move = false;
    while (curr != null) {
      ListNode next = curr.next;
      if (curr.val >= x && !move) move = true;
      // 将curr节点移动到链表的最前端
      if (curr.val < x && move) {
        // 当前位置删除curr节点
        prev.next = next;
        // curr节点插入到链表最前面
        curr.next = dummy.next;
        dummy.next = curr;
      } else prev = curr;
      curr = next;
    }
    return dummy.next;
  }
}