1. 程式人生 > >LintCode 刪除排序連結串列中的重複元素

LintCode 刪除排序連結串列中的重複元素

樣例

給出 1->1->2->null,返回 1->2->null

給出 1->1->2->3->3->null,返回 1->2->3->null

方法一:很容易就想到一種遞迴的演算法,從頭結點head開始,比較head和後一個節點的大小,如果相等,則把後一個節點刪掉。然後繼續比較頭結點和下一個節點。直到head和下一個節點不相同,遞迴呼叫函式,比較下一個節點和其後面節點的值。直到連結串列尾部。

遞迴程式碼如下:

class Solution {
public:
    /**
     * @param head: The first node of linked list.
     * @return: head node
     */
    ListNode *deleteDuplicates(ListNode *head) {
        // write your code here
        ListNode *p;
        p = head;
        if ( head == NULL ) {     //如果表為空表,返回NULL
            return NULL;
        }
        while ( p -> next != NULL ) {    // 魯棒性的表現
        if ( p ->val == p -> next -> val ) {    //如果head的值和下一個節點的值相同
            p -> next = p -> next -> next;      //將head的下一個節點刪除
            ListNode *deleteDuplicates(ListNode *head);   //繼續判斷
        } else {
        p = p -> next;      // 如果head和下一個值不想等,指標向後移動,對下一節點進行判斷
        ListNode *deleteDuplicates(ListNode *p);
        }
        }
        return head;
    }
};
方法二:關於連結串列很容易想到使用2個指標(快慢指標)。定義2個指標p,q。剛開始都讓他們指向頭節點head。然後讓q指向下一個節點,比較p,q節點的val值,如果相等,將q節點後移,刪除原來的q節點,繼續比較p,q;如果不相等,同時將p、q後移,繼續比較p、q的val值。直到q為空。(這裡迴圈結束的條件要注意,雖然q一直在向後移(q = q -> next),不能將迴圈結束的條件判斷為while(q -> next != NULL),如果這樣設定,在這麼一種情況下:1 1 2 5 5,最後時刻p指向倒數第二個5,q指向倒數第一個5,此時q ->next = NULL,迴圈跳出,輸出1 2 5 5 ,並沒有完成任務)。

程式碼如下:

 ListNode *p,*q;
        p = head;
        q = head;
        if ( head == NULL ) {
            return NULL;
        }
        q = p -> next;
        while ( q != NULL ) {    //迴圈判斷條件
            if ( p -> val == q -> val ) {
                q = q -> next;
                p -> next = p -> next -> next;
            } else {
                p = p -> next;
                q = q -> next;
            }
        }
        return head;