链表面试-链表分割(常考题)

题目描述:

编写代码,以给定值x为基准将链表分割成两部分,所有小于x的节点排在大于或等于x的节点之前,给定一个链表的头指针ListNode* pHead,请返回重新排列后的链表的头指针。注意:分割以后保持原来的数据顺序不变。

示例描述:

x=12,原链表如下:
链表面试-链表分割(常考题)
两个线段存放节点过程:
链表面试-链表分割(常考题)

分割后的链表如下:
链表面试-链表分割(常考题)

个人理解:

  • 本题用两个线段来存放比x小的节点和大于等于x的节点,刚开始的时候两个线段的头尾bs,be;as,ae;先置为null
  • 定义一个cur,在原来的链表中进行遍历
  • 如果cur.val<x,那么将该节点放到第一个线段,否则放到第二个线段,如果第一个线段没有数据,那么直接返回as;如果第一个线段有数据,那么将be和as进行拼接,be.next = as。
  • 当cur为null的时候,原来的单链表就遍历完了

题解实现:

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) {
        ListNode bs = null;//将两段的首尾先置为null
        ListNode be = null;
        ListNode as = null;
        ListNode ae = null;
        ListNode cur = pHead;
        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;
        }
        //1.判断bs是否为null,如果bs==null,则返回as
        if(bs == null){
            return as;
        }
        //2.如果bs不为null,需要进行拼接
        be.next = as;
        //3.如果ae不为null,那么ae的next需要置为null,此时返回bs
        if(ae!=null){
            ae.next = null;
        }
        return bs;
    }
}
上一篇:python爬虫库的补充


下一篇:Bootstrap简书(总结)