牛客網--鏈表分割

題目描述
編寫代碼，以給定值x爲基準將鏈表分割成兩部分，所有小於x的結點排在大於或等於x的結點之前
給定一個鏈表的頭指針 ListNode* pHead，請返回重新排列後的鏈表的頭指針。注意：分割以後保持原來的數據順序不變。

思路：
1.我們可以定義兩個鏈表（結構體類型變量）(比如small，large)，分別放小於x的結點、大於或等於x的結點。
2.對原鏈表進行遍歷，與x進行比較大小，對兩個新鏈表進行尾插（所以還要分別定義兩個指針哦）
3.遍歷完後，結點位置都已經放好了，我們再把兩個鏈表連起來就可以了
注：

這裏要用到 . 操作符，它是結構體類型變量訪問內部成員的；而 -> 是通過指針訪問結構體內部的變量

struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

class Partition {
public:
	ListNode* partition(ListNode* pHead, int x) {
		//先給兩個鏈表，定義結構體類型的變量
		ListNode small(0);
		ListNode large(0);
		//給兩個鏈表分別定義指針（進行尾插）
		ListNode* largeNode = &large;
		ListNode* smallNode = &small;
		while (pHead)
		{
			if (pHead->val < x)
			{
				smallNode->next = pHead;
				smallNode = pHead;
			}
			else
			{
				largeNode->next = pHead;
				largeNode = pHead;
			}
			pHead = pHead->next;
		}
		largeNode->next = NULL;//要讓最後的結點指向空
		smallNode->next = large.next;//將兩個鏈表連接
		return small.next;
	}
};