分割链表

题目描述：
编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或等于x的结点之前 给定一个链表的头指针ListNode* pHead，请返回重新排列后的链表的头指针。注意：分割以后保持原来的数据顺序不变

/*
遍历整个链表，把小于 x 的尾插到一个小链表，把大于等于 x 的尾插到一个大链表中
        理想情况下，把大链表接到小链表后边
        1) 如果没有小链表，直接返回大链表
        保证，返回链表的最后一个结点.next == null
        lessLast / greatLast
         */
/* 尾插
        1. 先情况讨论：
            1）如果当前链表为空，要插入的结点就是链表的第一个节点
            2) 如果链表不为空，
                1. 先找到当前的最后一个节点
                2. 让当前的最后一个结点的 next = 要插入的节点
                3. 如果每次的最后一个结点都是我们插入的
                    可以记录上次插入的最后一个节点
                4. 不要忘记更新最后一个节点
*/
class Patiton{
	public ListNode partition(ListNode pHead,int x){
		if(pHead==null||pHead.next==null){
			return pHead;
		}
		ListNode great = null;//大链表的头节点
		ListNode greatLast = null;//大链表的尾节点
		ListNode less = null;
		ListNode lessLast = null;
		ListNode cur = pHead;
		while(cur!=null){
			//尾插到小链表
			if(cur.val < x){
				if(less==null){
					less = cur;
				}else{
					lessLast.next = cur;
				}
				lessLast = cur;
			}else{
				//同理，尾插到大链表
				if(great==null){
					great = cur;
				}else{
					greatLast.next = cur;
				}
				greatLast = cur;
			}
			cur = cur.next;
		}
		if(less==null){
			return great;
		}else{//great不可能为空，至少有值为 x 的节点，所以不用讨论 great==null 的情况
			lessLast.next = great;
			if(greatLast!=null){//if(greatLast==null) 说明大链表中只有值为 x 的节点
				greatLast.next = null;
			}
			return less;
		}
	}
}