将单链表进行分区

2019独角兽企业重金招聘Python工程师标准>>>

原题

　　Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.
　　You should preserve the original relative order of the nodes in each of the two partitions.
　　For example,
　　Given 1->4->3->2->5->2 and x = 3,
　　return 1->2->2->4->3->5.

题目大意

　　给定一个单链表和一个值x，将链表分成小于等于x的部分和大于x的部分。同时保持链表原来的相对顺序。

解题思路

　　创建两个链表a，b，将原来链表中的每个结点，小于等于x的结点放在a链表的末尾，如果是大于就放在b的末尾，最后将b的头结点接到a末尾。

代码实现

结点类

public class ListNode {
    int val;
    ListNode next;
    ListNode(int x) { val = x; }
}

• 1
• 2
• 3
• 4
• 5

• 1
• 2
• 3
• 4
• 5

算法实现类

public class Solution {
    public ListNode partition(ListNode head, int x) {

        ListNode le = new ListNode(0); // 小于x的链表
        ListNode ge = new ListNode(0); // 大于等于x的链表
        ListNode t1 = le;
        ListNode t2 = ge;
        ListNode p = head;

        while (p != null) {
            if (p.val < x) {
                t1.next = p;
                t1 = p;
            } else {
                t2.next = p;
                t2 = p;
            }
            p = p.next;
        }

        t2.next = null;

        // 说明小于的链表上有数据
        if (t1 != le) {
            t1.next = ge.next;
            head = le.next;
        } else {
            head = ge.next;
        }
        return head;
    }

}

将单链表进行分区

 

转载于:https://my.oschina.net/u/2822116/blog/808351