Remove Duplicates from Sorted List II 解答(有个比较特殊的case leetcode OJ没有覆盖)

2 篇文章 0 订阅
1 篇文章 0 订阅

         昨天被考了一道数据结构题,当时的实现比较一般。回来翻看leetcode,果然是上面的题,遂解之。accept之后翻看discuss别人的解法,发现很多可以accept的代码都过不了我设计的一个case,网上搜了一些别人的代码,也过不了。遂有这篇博客,与君交流

         题目是这样的:
        Given a sorted linked list, delete all nodes that have duplicate numbers, leaving onlydistinct numbers from the original list.
         For example,Given 1->2->3->3->4->4->5, return 1->2->5.Given 1->1->1->2->3, return 2->3.

        

        当时拿到这个题目,一眼首先想到的就是类似于手游《糖果粉碎传奇》的连续消除的画面。一个比较强的case浮入脑海:
        输入:1->2->3->3->3->5->6->2->2->6->5->2->7
        输出:1->7

        分析:当3 和 第一段连续的2 被消除后,链表变成1->2->5->6->6->5->2->7;
                    此时出现了连续的6,遂需要消除6,结果是1->2->5->5->2->7;
                    同理消除5和2之后剩下1->7

         但是无论是leetcode里面的discuss还是网上搜到的别人可以accept的代码,都不是上面的这种输出,得到的输出一般如下:
         输出1:1 2 5 6 6 5 2 7
         输出2:1 2 5 5 2 7 
         当然,不知道是题目表意不明,或者是case太弱,但是题目明显写着‘distinct’,当然我认为我的case应该是对的,遂我来讲一下我的思路和做法。就拿 1->2->3->3->3->5->6->2->2->6->5->2->7 这个case举例。

         

        在这个题目中,链表是单链表,那么我们用指针遍历链表的时候,是无法回头的,这是个明显的限制条件。但是在这个case中,当删掉3和第一段连续2的之后链表变成1->2->5->6->6->5->2->7。不论你的代码怎么实现,此时你的指针可能在第一个6或者第一个5那里,当链表继续删除连续节点的时候,指针是不得不回头的,否则不能够删除出现的两个5的连续节点。既然这样,题目的解法基本思路可以确定:递归或者栈,当然,两者是想通的,不过栈的实现会方便一些(当时做这个题的时候硬着头皮写递归,没写好,哈哈)。

        用堆的方式解题,思路是比较简单的:
1、将头节点放入栈中,设置变量flag=false(flag表示栈顶节点是否需要被删除),检查下个节点
2、如果当前节点是NULL,则跳到第4步;
      如果当前节点跟栈顶节点相同,则跳过链表上这个节点,flag=true,继续执行第2步;
      如果跟栈顶节点值不同,则跳到第3步;
3、如果flag是true,那么把栈顶节点pop掉,flag置false;
      如果flag是false,把当前节点压入栈中,检查下个节点;
      执行第2步;
4、如果flag是true,则pop掉栈顶节点
5、链表的检查到此结束,栈内的节点就是结果链表的逆序存储,重新组织一下便可
6、注意:栈只需要存节点的指针

        复杂度分析:时间复杂度可以简单分析一下,因为每个节点要么被push要么被丢掉,push也有可能被pop,一共就这三种操作,每种最多执行一次,所以我们可以认为时间复杂度是O(n)。空间复杂度,最坏情况下会把整个链表的指针都放到栈里面,最好情况下是结果链表的长度,所以空间复杂度是O(n)


最后是我的代码:

struct ListNode {
  int val;
  ListNode *next;
  ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
 public:
  ListNode* deleteDuplicates(ListNode* head) {
    if (head == NULL) return NULL;

    stack<ListNode *> stk;
    bool flag = false;
    ListNode * cur = head;
    while (cur != NULL) {
      if (stk.empty()) {
          stk.push(cur);
          cur = cur->next;
          flag = false;
      } else if (cur->val == stk.top()->val) {
        flag = true;
        // because we don't know how the program frame manage the memory,
        // so i didn't free the  duplicate ListNode
        cur = cur->next;
        continue;
      } else if (flag) {
        stk.pop();
        flag = false;
      } else {   
        stk.push(cur);
        cur = cur->next;
      }   
    }   
    if (flag) stk.pop(); 
    
    // the remaining node in the stack is the result
    ListNode * res = NULL;
    while(!stk.empty()) {
      ListNode* t = stk.top();
      stk.pop();
      if (res == NULL) {
        res = t;
        res->next = NULL;
      } else {
        t->next = res;
        res = t;
      }
    } 
    return res;
  }
};



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值