leetcode: Partition List

问题描述：

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.

原问题链接：https://leetcode.com/problems/partition-list/

 

问题分析

　　这个问题的思路其实比较简单，我们需要将一个链表按照某个给定的值给划分成两个部分。一个部分小于这个值，一个部分大于这个值。那么我们可以声明两个链表节点，一个保存小于这个值的元素，另外一个保存大于或者等于这个值的元素。

　　我们需要遍历这个链表，每次碰到小于值的元素，则将它放到小于这个元素的链表里，否则放入另外一个链表里。所以需要额外定义两个元素来专门跟踪这两个链表的尾部。在遍历完之后还需要将大的那个链表的最后一个元素的next设置为null。

　　详细的代码实现如下：

 

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ListNode partition(ListNode head, int x) {
        if(head == null || head.next == null) return head;
        ListNode lessNode = new ListNode(0), biggerNode = new ListNode(0);
        ListNode lessHead = lessNode, biggerHead = biggerNode, cur = head;
        while(cur != null) {
            if(cur.val < x) {
              lessHead.next = cur;
              lessHead = lessHead.next;
            } else {
                biggerHead.next = cur;
                biggerHead = biggerHead.next;
            }
            cur = cur.next;
        }
        biggerHead.next = null;
        lessHead.next = biggerNode.next;
        return lessNode.next;
    }
}

　　这里的代码实现并不复杂，只是很容易出错。很多细节需要注意。