LintCode 96.連結串列劃分
阿新 • • 發佈:2018-12-13
題目說明
思路
從頭遍歷,找到大於等於x的元素提出來建立一個新連結串列,再和原來的連結串列連線。 無法找到其前一個節點。
最終思路 head指標從頭到尾遍歷該連結串列的每一個元素 逐個與x進行比較
建立兩個頭結點
- 連結比x小的元素
- 連結比x大的元素 並通過比那裡使之串接起來
最後合併兩個連結串列
/**
* Definition of singly-linked-list:
* class ListNode {
* public:
* int val;
* ListNode *next;
* ListNode(int val) {
* this->val = val;
* this->next = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param head: The first node of linked list
* @param x: An integer
* @return: A ListNode
*/
ListNode * partition(ListNode * head, int x) {
// write your code here
ListNode *smallHead = new ListNode(0);
ListNode *bigHead = new ListNode(0);
ListNode *smallRear = smallHead;
ListNode *bigRear = bigHead;
while(head != NULL)
{
if(head->val < x)
{
smallRear->next = head;
smallRear = head;
}
else
{
bigRear->next = head;
bigRear = head;
}
head = head->next;
}
smallRear->next = bigHead->next;
bigRear->next = NULL;
return smallHead->next;
}
};
標準答案
/**
* 本參考程式來自九章演算法,由 @九章演算法 提供。版權所有,轉發請註明出處。
* - 九章演算法致力於幫助更多中國人找到好的工作,教師團隊均來自矽谷和國內的一線大公司在職工程師。
* - 現有的面試培訓課程包括:九章演算法班,系統設計班,演算法強化班,Java入門與基礎演算法班,Android 專案實戰班,
* - Big Data 專案實戰班,演算法面試高頻題班, 動態規劃專題班
* - 更多詳情請見官方網站:http://www.jiuzhang.com/?source=code
*/
/**
* Definition of ListNode
* class ListNode {
* public:
* int val;
* ListNode *next;
* ListNode(int val) {
* this->val = val;
* this->next = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param head: The first node of linked list.
* @param x: an integer
* @return: a ListNode
*/
ListNode *partition(ListNode *head, int x) {
ListNode *leftDummy = new ListNode(0);
ListNode *rightDummy = new ListNode(0);
ListNode *leftTail = leftDummy;
ListNode *rightTail = rightDummy;
while (head != NULL) {
if (head->val < x) {
leftTail->next = head;
leftTail = head;
} else {
rightTail->next = head;
rightTail = head;
}
head = head->next;
}
leftTail->next = rightDummy->next;
rightTail->next = NULL;
return leftDummy->next;
}
};