欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Leetcode 83: 删除排序链表中的重复元素

程序员文章站 2024-03-23 09:27:34
...

题目

给定一个排序链表,删除所有重复的元素,使得每个元素只出现一次。

示例 1:

输入: 1->1->2
输出: 1->2
示例 2:

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

分析:

算法
这是一个简单的问题,仅测试你操作列表的结点指针的能力。由于输入的列表已排序,因此我们可以通过将结点的值与它之后的结点进行比较来确定它是否为重复结点。如果它是重复的,我们更改当前结点的 next 指针,以便它跳过下一个结点并直接指向下一个结点之后的结点。

C

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* deleteDuplicates(struct ListNode* head) {
    struct ListNode *p;
    p = head;
    while(p!=NULL&&p->next!=NULL)
    {
        if(p->val==p->next->val)
        {
            p->next=p->next->next;
        }
        else
            p=p->next;
    }
    return head;
}

python

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def deleteDuplicates(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        p = head
        while p and p.next :
            if p.val==p.next.val:
                p.next=p.next.next
            else:
                p=p.next
        return head

总结

正确性

我们可以通过定义循环不变式来证明此代码的正确性。循环不变式是在循环的每次迭代之前和之后为真的条件。在这种情况下,一个帮助我们证明正确性的循环不变式是这样的:

列表中直到指针 current 的所有结点都不包含重复的元素。

我们可以用归纳法证明这种情况确实是循环不变式。在进入循环之前,current 指向列表的头部。因此,列表中直到 current 的部分只包含头部。因此它不能包含任何重复的元素。现在假设current 现在指向列表中的某个结点(但不是最后一个元素),并且列表中直到 current 的部分不包含重复元素。在另一个循环迭代之后,发生两件事之一。

current.next 是 current 的副本。在这种情况下,删除 current.next 中的重复结点,并且current保持指向与之前相同的结点。因此,情况仍然成立;一直到 current 仍然没有重复项。

current.next 不是 current 的副本(并且,因为列表已经排序,current.next 也不是 current 之前出现的任何其他元素的副本)。在这种情况下,current 向前移动一步指向 current.next。因此,情况仍然成立;一直到 current 仍然没有重复项。

在循环的最后一次迭代中,current 必定指向最后一个元素,因为再往后,current.next = null。因此,在循环结束后,直到最后一个元素的所有元素都不包含重复项。