Java刷题篇——链表分割

2023-12-20 13:24:03

链表分割

1. 题目描述

现有一链表的头指针 ListNode* pHead,给一定值x,编写一段代码将所有小于x的结点排在其余结点之前,且不能改变原来的数据顺序,返回重新排列后的链表的头指针。

2. 思路

  1. 判断链表是否为空,如果为空返回null

  2. 定于指针cur指向头节点,即链表的头节点

  3. 定义指针bsbe,分别指向小于x链表的头节点和尾节点

  4. 定义指针asae,分别指向大于等于x链表的头节点和尾节点

  5. 循环遍历链表,直到遍历完所有的节点

    • 如果当前节点的val值小于x

      1. 判断小于x的链表中是否为空,如果为空则bsbe都指向当前节点

      2. 否则benext指向当前节点,be指向benext

    • 当前节点的val值大于等于x

      1. 判断大于等于x的链表中是否为空,如果为空则asae都指向当前节点

      2. 否则aenext指向当前节点,ae指向aenext

    • cur指向curnext

  6. 如果小于x的链表为空,则返回as

  7. 否则benext指向as

  8. 如果as不为空,则将asnext值为空(原链表的最后一个节点的值小于x

  9. 返回bs

image-20231220090548125

3. 代码

import java.util.*;

/*
public class ListNode {
    int val;
    ListNode next = null;

    ListNode(int val) {
        this.val = val;
    }
}*/
public class Partition {
    public ListNode partition(ListNode pHead, int x) {
        if (pHead == null) {
            return null;
        }
        // write code here
        // 定义cur 指向头结点
        ListNode cur = pHead;

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

        while (cur != null) {
            if (cur.val < x) {
                if (bs == null) { // 判断是不是第一个元素
                    bs = cur;
                    be = cur;
                } else {
                    be.next = cur;
                    be = be.next;
                }
            } else {
                if (as == null) {
                    as = cur;
                    ae = cur;
                } else {
                    ae.next = cur;
                    ae = ae.next;
                }
            }
            cur = cur.next;
        }
        if (be == null) {
            return as;
        }
        be.next = as;
        if (as != null) {
            ae.next = null;
        }
        return bs;
    }
}

运行结果:

image-20231220090629696

文章来源:https://blog.csdn.net/2301_79076048/article/details/135099292
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。