1. 程式人生 > >LintCode 96.連結串列劃分

LintCode 96.連結串列劃分

題目說明

題目說明

思路

從頭遍歷,找到大於等於x的元素提出來建立一個新連結串列,再和原來的連結串列連線。 無法找到其前一個節點。

最終思路 head指標從頭到尾遍歷該連結串列的每一個元素 逐個與x進行比較

建立兩個頭結點
  1. 連結比x小的元素
  2. 連結比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;
    }
};