今天看啥  ›  专栏  ›  承志

LeetCode 82,考察你的基本功,在有序链表中删除重复元素II

承志  · 掘金  ·  · 2020-07-13 03:16
阅读 96

LeetCode 82,考察你的基本功,在有序链表中删除重复元素II

本文始发于个人公众号:TechFlow,原创不易,求个关注


今天是LeetCode专题的第51篇文章,我们来看LeetCode第82题,删除有序链表中的重复元素II(Remove Duplicates from Sorted List II)。

这题官方给出的难度是Medium,点赞1636,反对107,通过率在36.3%左右。根据我们之前的分析,这题的难度适中,并且质量很高,好评如潮。实际上也的确如此,这题算法本身并不难,但是想要完整没有bug地实现并不容易,我们一起来看看。

题意

给定一个有序的存在重复元素的链表,要求移除掉链表当中所有的重复元素。返回一个不包含重复元素的链表。

这里要注意的一点,这题让我们做的事情并不是去重,就是去除掉多余的元素,而是要去除掉所有重复的元素。比如2在链表当中出现了两次,属于重复元素,我们要做的并不是去掉一个2,仅保留一个,而是要将所有的2都去除,因为2属于重复元素。

我们来看样例:

Input: 1->2->3->3->4->4->5
Output: 1->2->5
复制代码

原链表当中的3和4都属于重复元素,所以被去除了。

Input: 1->1->1->2->3
Output: 2->3
复制代码

解法

前面说了这题的质量很高,这题是属于典型的解法赤裸裸,但是很多人就是写不出来的题,非常考察基本功。适合用在校招面试当中,如果我有幸去面试校招生, 我可能会选这道题。不存在算法会不会的问题,写不出来一定是基本功不够扎实。

链表已经有序了,那么相同的元素必然会排在一起,我们只需要将它们移除就可以了。但是说起来简单,要在链表当中实现并不容易。难点主要有两个,一个是链表增删节点的操作很多人不熟悉,尤其是像是C++这样的语言涉及指针,可能更不容易。另外一个难点就在题意当中,我们要做的不是去重,而是要所有重复的元素全部删除

看起来似乎和去重没什么差别, 如果你真这么想,并且着手去实现,那么几乎可以肯定一定会遇到问题。

因为链表是单向的,假设你当前的指针是cur,当你发现cur这个指针的元素存在重复的时候,你需要连当前这个节点一起删除。我们都知道,单向链表是不能走回头路的,而删除节点,必须要用到前一个节点的指针。再加上判断元素重复需要用到的指针,会需要我们同时维护多个指针,增加代码的编码难度。

针对这个问题,我们有两种解决思路。第一种是我们不在原链表上处理,而是创建一个新的链表进行返回。所以我们要做的就不是删除元素,而是插入元素,只有发现当前元素不存在重复的时候才会插入链表。最后返回的也是一个全新的链表。

这当然是可以的,也是没有问题了,我第一次做这道题就是采取的这种措施。相比之下, 这种方法会容易一些,因为我们不需要判断太多的指针和位置,我贴一下当时的代码:

class Solution {
public:
    ListNode* deleteDuplicates(ListNode* head) {
          // 如果元素少于2个则直接返回
        if (head == nullptr || head->next == nullptr) return head;
        int cur = head->val;
          // 初始化
        bool flag = false;
        ListNode* ct = new ListNode(0);
        ListNode* pnt = ct;
        head = head->next;
          // 遍历元素
        while (head) {
            int hd = head->val;
               // 判断当前元素与之前的元素是否相等
            if (hd != cur) {
                   // 之前的元素没有出现重复
                if (!flag) {
                       // 把之前的元素存入链表
                    pnt->next = new ListNode(cur);
                       // 链表移动
                    pnt = pnt->next;
                }else flag = false;
                   // 更新之前的元素
                cur = hd;
            }else flag = true;
            head = head->next;
        }
          // 单独处理最后一个元素
        if (!flag) pnt->next = new ListNode(cur);
        return ct->next;
    }
};
复制代码

虽然题目当中没有对解法做出限制,也没有规定我们必须要在原链表上进行处理,但是这种创建新链表的方法终归有绕开问题的嫌疑。所以我们还有第二种解法,就是直面问题,我们维护多个指针,判断当前位置的下一个元素是否构成重复。如果重复,则删除掉重复的部分。

正如我们之前所说的那样,在单向链表当中很难删除当前元素,所以我们判断下一个元素是否会构成重复。如果重复的话,进行删除要可行许多。这样也有一个问题就是,有可能链表的第一个元素就是重复的,我们没有办法找到第一个元素的上一个元素。针对这个问题,我们采用的方法是人为给它创造一个元素放在首元素之前。这样我们整个流程就可以串起来了,唯一的难点就是编码了。

我们仔细一些,写出代码还是可以的:

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def deleteDuplicates(self, head: ListNode) -> ListNode:
        # 创建一个新的元素放在链表头部,这样原本第一个元素就是头指针的下一个元素了
        node = ListNode()
        node.next = head
        # pnt指向新创建的元素
        pnt = node
        
        while pnt.next is not None:
            # cur指向当前位置的下一个位置
            # 我们要做的即判断cur这个指针的元素是否重复
            cur = pnt.next
            if cur.next is None:
                break
             
            # ptr指向cur的next
            ptr = cur.next
            # 如果ptr和cur相等,那么出现重复,我们需要跳过所有相等的元素
            if ptr is not None and ptr.val == cur.val:
                while ptr is not None and ptr.val == cur.val:
                    ptr = ptr.next
                pnt.next = ptr
            # 否则说明不重复,移动pnt
            else:
                pnt = pnt.next
                
        # 由于我们开始的时候人为添加了一个辅助元素
        # 返回的时候要将它去除
        return node.next
复制代码

总结

这道题的算法很简单,我想大部分人都能想出解法来,但是要将解法实现并不太容易。这当中用到了很多小技巧,比如我们认为创建了一个新的头结点,比如我们将删除当前元素转化成了删除下一个元素等等。这些技巧虽然算不上什么,但是灵活使用,可以大大降低我们编码的复杂度,也正是因为这一点,这题的质量非常高,值得一做。

很多人非常讨厌涉及链表的问题,觉得链表很难操作,容易写错,但实际上这是基本功的很重要的一部分。很多公司喜欢考察候选人的基本功,提升这方面的能力对于我们应聘或者是工作非常有帮助。

今天的文章到这里就结束了,如果喜欢本文的话,请来一波素质三连,给我一点支持吧(关注、转发、点赞)。

本文使用 mdnice 排版




原文地址:访问原文地址
快照地址: 访问文章快照