【Leetcode】86. 分隔链表

题目描述

给你一个链表的头节点 head 和一个特定值 x ，请你对链表进行分隔，使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。

你应当 保留 两个分区中每个节点的初始相对位置。

题解

非常简单粗暴又有效，让small用于链接小结点，large用于链接大结点，最后两个结点链接起来。

执行用时：0 ms, 在所有 Java 提交中击败了100.00%的用户

内存消耗：37.6 MB, 在所有 Java 提交中击败了86.29%的用户

通过测试用例：168 / 168

import java.util.ArrayList;
import java.util.List;

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode partition(ListNode head, int x) {
        ListNode small = new ListNode(0);  // small专于小结点
        ListNode small_head = small;
        ListNode large = new ListNode(0);  // large专于大结点
        ListNode large_head = large;
        while (head != null) {  // head移动，遍历链表
            if (head.val < x) {  // 发现小于x的节点，让small指针链接
                small.next = head;
                small = small.next;
            }
            else {  // 发现大于x的节点，让large指针链接
                large.next = head;
                large = large.next;
            }
            head = head.next;
        }
        large.next = null;
        small.next = large_head.next;  // small和large两个链表链接
        return small_head;
    }
}