链表面试题（6）

最新推荐文章于 2022-03-20 11:52:52 发布

胡歌女票

最新推荐文章于 2022-03-20 11:52:52 发布

阅读量114

点赞数

分类专栏： java

本文链接：https://blog.csdn.net/zxjkl123/article/details/106653063

版权

java 专栏收录该内容

80 篇文章 0 订阅

订阅专栏

（6）、分割链表
编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或等于x的结点之前
给定一个链表的头指针 ListNode* pHead，请返回重新排列后的链表的头指针。注意：分割以后保持原来的数据顺序不变。
https://www.nowcoder.com/practice/0e27e0b064de4eacac178676ef9c9d70?

import java.util.*;
public class Partition {
    public ListNode partition(ListNode pHead, int x) {
        ListNode bs = null;
        ListNode be = null;
        ListNode as = null;
        ListNode ae = null;
        ListNode cur = pHead;
        while (cur != null) {
            //判断是否小于x
            if (cur.val < x) {
                //判断是不是第一次插入
                if (bs == null) {
                    bs = cur;
                    be = bs;
                } else {
                    be.next = cur;
                    be = be.next;
                }
            } else {
                //判断是不是第一次插入
                if (as == null) {
                    as = cur;
                    ae = as;
                } else {
                    ae.next = cur;
                    ae = ae.next;
                }
            }
            cur = cur.next;
        }
        //第一个区间没有数据
        if (bs == null) {
            return as;
        }
        //将两个区间连接
        be.next = as;
        //将as.next置为空
        if (as != null) {
            ae.next = null;
        }
        return bs;
    }
}