按照给定值划分链表

描述：给出一个长度为 nn 的单链表和一个值 xx ，单链表的每一个值为 list[i]list[i]，请返回一个链表的头结点，要求新链表中小于 xx 的节点全部在大于等于 xx 的节点左侧，并且两个部分之内的节点之间与原来的链表要保持相对顺序不变。

例如：
给出 1→4→3→2→5→2 和 x=3
返回 1→2→2→4→3→5

复杂度要求：
时间 O(n)
空间 O(1)

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
        // 整体思路：将链表拆成两部分，一部分比x小，另一部分大于等于x，最后将两个链表拼接
        ListNode h1 = null;
        ListNode h2 = null;
        // h3表示小于x的链表头结点
        ListNode h3 = null;
        // h4表示大于等于x的链表头结点
        ListNode h4 = null;
        ListNode p = head;
        ListNode q = null;
        while(p!=null){
            q = p.next;
            if(p.val<x){
                if(h1==null){
                    h1=p;
                    h3=p;
                    // 注意链表尾部必须要置空，否则会进行无限循环
                    p.next = null;
                }else{
                    h1.next=p;
                    h1=p;
                    p.next = null;
                }
            }else{
                if(h2==null){
                    h2=p;
                    h4=p;
                    p.next = null;
                }else{
                    h2.next=p;
                    h2=p;
                    p.next = null;
                }
            }
            p=q;
        }
        if(h1!=null){
            h1.next=h4;
        }
        if(h3!=null){
            return h3;
        }
        return h4;
    }
    
    
    
}