题目:给出一个链表和一个值 ,以 X为参照将链表划分成两部分,使所有小于X 的节点都位于大于或等于X 的节点之前。
两个部分之内的节点之间要保持的原始相对顺序。
例如:
给出 1→4→3→2→5→2 和 x=3,
返回 1→2→2→4→3→5.
解答:定义两个链表p1,p2,从头遍历head,大于等于x值的放入p2,小于x值的放入p1,最后把p1的最后节点的next节点执向p2的头节点。
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* }
*/
public class Solution {
/**
*
* @param head ListNode类
* @param x int整型
* @return ListNode类
*/
public ListNode partition (ListNode head, int x) {
// write code here
if(head == null || head.next == null) return head;
ListNode left = new ListNode(0);
ListNode right = new ListNode(0);
ListNode cur = head;
ListNode p1 = left;
ListNode p2 = right;
while(cur != null){
if(cur.val >= x){
p2.next = cur;
p2 = cur;
}else{
p1.next = cur;
p1 = cur;
}
cur = cur.next;
}
p2.next = null;
p1.next = right.next;
return left.next;
}
}