Leetcode86.分隔链表
题目连接:86.分隔链表
思路:遍历一遍链表,将表中结点分为两类,一类是小于给定的x,另一类是大于等于x。输出结果时要考虑四种情况:输入的表头为空;没有小于x的节点;没有大于等于x的节点;既有小于x又有大于等于x的。
代码:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode partition(ListNode head, int x) {
if(head == null) return head;
int len = 0, li = 0, mti = 0;
for(ListNode p=head; p!=null; p=p.next) len ++;
ListNode[] less = new ListNode[len];
ListNode[] moreThen = new ListNode[len];
for(ListNode p=head; p!=null; p=p.next){
if(p.val < x){
if(li > 0) less[li-1].next = p;
less[li++] = p;
}else{
if(mti > 0) moreThen[mti-1].next = p;
moreThen[mti++] = p;
}
}
if(mti>0 && li>0){
less[li-1].next = moreThen[0];
moreThen[mti-1].next = null;
return less[0];
}
if(li==0){
return moreThen[0];
}
return less[0];
}
}
执行用时:0 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗:37.3 MB, 在所有 Java 提交中击败了98.07%的用户