LeetCode——86. 分隔链表(链表)

86. 分隔链表(链表)

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/partition-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目**

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

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

示例:

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


思想**

分别建立以虚节点dummyNodeL和dummyNodeR为头结点的链表,为之分别加入原链表中值小于和大于x的结点;
再连接两个链表,更新头结点返回即可

代码**

/**
 * 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) {
        if(head == null || head.next == null){//特殊边界处理
            return head;
        }
        //分别建立以虚节点dummyNodeL和dummyNodeL为头结点的链表分别记录小于和大于x的数
        ListNode dummyNodeL = new ListNode(-1);
        ListNode dummyNodeR = new ListNode(-1);
        ListNode left = dummyNodeL;
        ListNode right = dummyNodeR;
        while(head != null){
            if(head.val < x){
                left.next = head;
                left = left.next;
            }
            else{
                right.next = head;
                right = right.next;
            }
            head = head.next;
        }
        right.next = null;//不加该句会使得链表存在循环
        //连接两个链表
        left.next = dummyNodeR.next;
        head = dummyNodeL.next;
        return head;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_34767784/article/details/107279543
今日推荐