算法|力扣题解(面试题 02.04. 分割链表)

题目

【算法|力扣题解(面试题 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)

    推荐阅读