链表分割——牛客剑指offer
题目描述:
编写代码,以给定值x为基准将链表分割成两部分,所有小于x的结点排在大于或等于x的结点之前
给定一个链表的头指针 ListNode pHead,请返回重新排列后的链表的头指针。注意:分割以后保持原来的数据顺序不变。
题目分析:
将链表分割为两部分,以x值为分割线(x不一定存在于链表中):前部分链表结点的值是小于x的,而且这一部分各结点的顺序与在原链表中的相对顺序相同。后部分链表结点的值是大于等于x的,而且这一部分各结点的顺序与在原链表中的相对顺序相同。
例如,链表{1,3,7,4,2,5} 以值4分割,结果为{1,3,2,4,7,5}
链表{1,3,7,2,5} 以值4分割,结果为{1,3,2,7,5}
解题思路:
新建一个链表-->创建一个值为x的结点nodex
迭代遍历原链表的结点:
如果结点值小于x:插入到x结点nodex的前面(插入到前半部分的最后一个位置)
如果结点指大于x:插入到新链表的最后面(插入到后半部分的最后一个位置)
代码实现:(含验证)
1 class ListNode {
2 int val;
3 ListNode next = null;
4
5 ListNode(int val) {
6 this.val = val;
7 }
8 }
9 public class splitList {
10 public ListNode partition(ListNode pHead, int x) {
11 if(pHead==null){
12 return null;
13 }
14 //创建x结点作为分割前半部分和后半部分的中间结点
15 ListNode nodex=new ListNode(x);
16 //创建newHead结点方便第一个小于x值的插入
17 ListNode newHead=new ListNode(0);
18 newHead.next=nodex;
19 //创建before结点,在迭代过程中始终保持before.next=nodex
20 //从而保证小于x值的结点可以插入到nodex结点之前
21 ListNode before=newHead;
22 //创建after结点,在迭代过程中始终保持after结点是最后一个结点
23 //从而保证大于等于x值的结点可以插入链表的最后位置
24 ListNode after=nodex;
25 ListNode walkNode=pHead;
26 boolean isexist=false;
27 while(walkNode!=null){
28 //如果当前节点小于x,复制结点并将其插入到xnode的前一个结点,然后移动before指针
29 if(walkNode.val<x){
30 ListNode node=new ListNode(walkNode.val);
31 before.next=node;
32 node.next=nodex;
33 before=node;
34 }
35 //如果当前节点大于x,复制结点并将其插入到链表最后一个结点,然后移动after指针
36 else if(walkNode.val>=x){
37 ListNode node=new ListNode(walkNode.val);
38 after.next=node;
39 after=node;
40 }
41 walkNode=walkNode.next;
42 }
43 //忽略自建的x结点nodex和头结点newHead;
44 //因为x结点不一定存在于原链表,所以此处要将分开的前后部分相连
45 before.next=nodex.next;
46 return newHead.next;
47 }
48 public static void main(String []args){
49 ListNode head = new ListNode(3);
50 ListNode node2 = new ListNode(3);
51 ListNode node3 = new ListNode(3);
52 ListNode node4 = new ListNode(3);
53 head.next = node2;
54 node2.next = node3;
55 node3.next = node4;
56 splitList split =new splitList();
57 ListNode returnHead=split.partition(head,3);
58 while (returnHead != null)
59 {
60 System.out.println(returnHead.val );
61 returnHead = returnHead.next;
62 }
63 System.out.println();
64 }
65 }