面试题02.04分割链表

链表分区算法实现

原创已于 2022-03-15 14:25:46 修改 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#链表 #leetcode #数据结构

于 2022-03-15 14:25:16 首次发布

算法题专栏收录该内容

33 篇文章

订阅专栏

该博客详细介绍了如何实现一个链表分区的算法，将链表中的节点根据值小于或大于给定的整数x分成两部分。代码中使用了双指针技巧，分别维护两个链表，一个用于存储小于x的节点，另一个用于存储大于x的节点。最后，将小于x的链表接在大于x的链表之后，并处理特殊情况，如没有节点小于x的情况。整个过程保持了节点原有的顺序。

题目描述：

在这里插入图片描述

思路：

需要将一个链表分为两部分，前部分小于x，后部分大于x。并且节点之前的顺序不能变。所以每次向新链表添加时需要进行尾插法。分为两部分，值大于x的分为一部分，小于x的分为一部分。使用as，ae来表示大的那一部分的头和尾节点。使用bs，be来表示小的那一部分。使用中间变量cur来代替head遍历链表。当第一次插入节点时，使得bs/as=cur，并且be/ae=cur。非第一次插入则让be/ae.next=cur，be/ae=be/ae.next，be/ae始终保持在链表尾部。当所有的节点遍历完后，需要对bs进行判断，如果没有任何节点小于x，则返回as即可。注意：如果as不为空，需要将ae.next置为空，若ae.next不为空，可能造成死循环。
在这里插入图片描述

代码：

/**
 * 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){
                return null;
            }

            ListNode bs=null;
            ListNode be=null;
            ListNode as=null;
            ListNode ae=null;


            ListNode cur=head;


            while(cur!=null){
                if(cur.val<x){
                    if(bs==null){
                        bs=cur;
                        be=cur;
                    }else{
                        be.next=cur;
                        be=cur;
                    }
                    cur=cur.next;
                }else{
                    if(as==null){
                        as=cur;
                        ae=cur;
                    }else{
                        ae.next=cur;
                        ae=cur;
                    }
                    cur=cur.next;
                }
            }

            if(bs==null){
                return as;
            }

            be.next=as;

            if(as!=null){
                ae.next=null;
            }

            return bs;
    }

}