leetcode86 分隔链表

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

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

示例:输入: head = 1->4->3->2->5->2, x = 3  

        输出: 1->2->2->4->3->5

把所有小于给定值的节点都移到前面,大于该值的节点顺序不变。将所有小于给定值的节点取出组成一个新的链表,此时原链表中剩余的节点的值都大于或等于给定值,只要将原链表直接接在新链表后即可。

public ListNode partition(ListNode head, int x) {
    if(head == null) return head;
    ListNode dummy = new ListNode(-1);
    ListNode newDummy = new ListNode(-1);
    dummy.next = head;
    ListNode cur = dummy;
    ListNode p = newDummy;
    while(cur.next != null){
        if(cur.next.val < x){
            p.next = cur.next;
            p = p.next;
            cur.next = cur.next.next;
            p.next = null;
        }
    }
    p.next = dummy.next;
    return newDummy.next;
}

猜你喜欢

转载自blog.csdn.net/qq_43322057/article/details/84670618