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

题目

给你一个链表的头节点 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;
  }
}

题目来源：力扣（LeetCode）