链表分割——牛客剑指offer

题目描述:

编写代码,以给定值x为基准将链表分割成两部分,所有小于x的结点排在大于或等于x的结点之前

给定一个链表的头指针 ListNode pHead,请返回重新排列后的链表的头指针。注意:分割以后保持原来的数据顺序不变。

题目分析:

将链表分割为两部分,以x值为分割线(x不一定存在于链表中):前部分链表结点的值是小于x的,而且这一部分各结点的顺序与在原链表中的相对顺序相同。后部分链表结点的值是大于等于x的,而且这一部分各结点的顺序与在原链表中的相对顺序相同。

例如,链表{13,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 }

猜你喜欢

转载自www.cnblogs.com/darlinFly/p/9343388.html