给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
答案:
1public ListNode partition(ListNode head, int x) {
2 ListNode smallerHead = new ListNode(0);
3 ListNode biggerHead = new ListNode(0);
4 ListNode smaller = smallerHead;
5 ListNode bigger = biggerHead;
6 while (head != null) {
7 if (head.val < x) {
8 smaller = smaller.next = head;
9 } else {
10 bigger = bigger.next = head;
11 }
12 head = head.next;
13 }
14 smaller.next = biggerHead.next;
15 bigger.next = null;
16 return smallerHead.next;
17}
解析:
看变量名就能明白,就不再介绍。