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

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

題目描述:

給定一個排序連結串列,刪除所有重複的元素,使得每個元素只出現一次。

示例 1:

輸入: 1->1->2
輸出: 1->2

示例 2:

輸入: 1->1->2->3->3
輸出: 1->2->3

C++程式碼:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* deleteDuplicates(ListNode* head) {
        
        ListNode *p = head, *temp = NULL;
        if (head == NULL) return head;
        while (p->next != NULL){
            if (p->val == p->next->val){
                temp = p->next;
                p->next = p->next->next;
                free(temp);
            }
            else
                p = p->next;
        }
        
        return head;
    }
};