跳转至

86.分隔链表

题目描述

原题

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

提示:

  • 链表中节点的数目在范围 [0, 200]
  • -100 <= Node.val <= 100
  • -200 <= x <= 200

题解

/**
 * 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; }
 * }
 */
//思路很简单先遍历一遍原始的链表删掉那些小于x的节点,可以参考203题
//把删除的节点串成一个链表,最后两个链表合并成一个链表即可 
class Solution {
    public ListNode partition(ListNode head, int x) {
        //小于x的节点
        ListNode dummyNode1 = new ListNode();
        //大于等于x的节点
        ListNode dummyNode2 = new ListNode();
        dummyNode2.next = head;
        ListNode curr1 = dummyNode1;
        ListNode curr2 = dummyNode2;
        while(curr2.next!=null){
            if(curr2.next.val < x){
                curr1.next = curr2.next;
                curr1 = curr1.next;
                curr2.next = curr2.next.next;
            }else{
                curr2 = curr2.next;
            }
        }
        curr1.next = dummyNode2.next;
        return dummyNode1.next;
    }
}