LeetCode147:Insertion Sort List

mickole發表於2014-02-17

題目:

Sort a linked list using insertion sort.

解題思路:

按題目要求,直接進行插入排序

實現程式碼:

#include <iostream>

using namespace std;
/*
Sort a linked list using insertion sort.
 */
struct ListNode
{
    int val;
    ListNode *next;
    ListNode(int x):val(x), next(NULL){}    
};

void addNode(ListNode* &head, int val)
{
    ListNode *newNode = new ListNode(val);
    if(head == NULL)
    {
        head = newNode;
    }
    else
    {
        newNode->next = head;
        head = newNode;
    }
}

void PrintList(ListNode *root)
{
    ListNode *head = root;
    while(head != NULL)
    {
        cout<<head->val<<"\t";
        head = head->next;
    }
    cout<<endl;
}

class Solution {
public:
    ListNode *insertionSortList(ListNode *head) {
        if(head == NULL || head->next == NULL)
            return head;
        ListNode *p = head->next;//儲存除第一個節點之後的所有節點,待之後逐個進行插入排序 
        head->next = NULL;       
        while(p)
        {
            ListNode *t = p->next;//儲存p節點的下一個節點 
            if(p->val < head->val)//如果當前要插入的節點小於頭節點 
            {
                
                p->next = head;
                head = p;//將頭結點指向新插入的節點           
            }
            else//否則從頭結點的下一個節點開始逐個與當前要插入節點比較,直到遇到大於當前節點的節點 
            {
                ListNode *q = head;
                while(q->next && q->next->val <= p->val)
                    q = q->next;
    
                p->next = q->next;
                q->next = p;    
            }
            p = t;
        }
        return head;                
    }
};

int main(void)
{
    ListNode *head = new ListNode(5);
    addNode(head, 3);
    addNode(head, 10);
    addNode(head, 15);
    PrintList(head);
    
    Solution solution;
    head = solution.insertionSortList(head);
    PrintList(head);
    
    return 0;
}

相關文章