Bootstrap

LeetCode 面试题 02.04. 分割链表

一、题目

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

二、C# 题解

  最初打算在原链表上改,想了很久发现难以操作,需要引入队列。最后决定,不如直接新建两个链表 smalllarge,分别用于添加节点值 < x < x <x 和节点值 ≥ x \geq x x 的节点。
遍历链表 head 后,拼接 smalllarge 链表,最终返回头节点 small.next 即可。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     public int val;
 *     public ListNode next;
 *     public ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ListNode Partition(ListNode head, int x) {
        ListNode small = new ListNode(0), large = new ListNode(0);
        ListNode p = small, q = large; // p 指向 small 尾端,q 指向 large 尾端

        while (head != null) {  // 遍历原链表
            if (head.val < x) { // 小值放入 small 链表中
                p.next = head;
                p = p.next;
            }
            else {
                q.next = head;  // 大值放入 large 链表中
                q = q.next;
            }

            head = head.next;
        }

        p.next = large.next;    // 连接两个链表
        q.next = null;          // 断后

        return small.next;
    }
}
  • 时间复杂度: O ( n ) O(n) O(n)
  • 空间复杂度: O ( n ) O(n) O(n)
;