[2021校招必看之Java版《剑指offer》-56] 删除链表中重复的结点

1、题目描述

  【JZ56】在一个排序的链表中，存在重复的结点，请删除该链表中重复的结点，重复的结点不保留，返回链表头指针。
  例如，链表1->2->3->3->4->4->5 处理后为 1->2->5
  知识点：单链表
  难度：☆

2、解题思路

2.1 暴力遍历

  遍历链表，用一个 HashSet 把重复的结点保存起来，然后再遍历链表，把存在于 HashSet 的链表删除即可。

2.2 直接删除

  先定义一个新的根结点 root ，值为 -1。再定义一个当前正在遍历的结点 cur，定义一个当前结点的前一个结点 pre。

  初始化：root.next = pHead; pre = root; cur = root.next;

  在遍历单链表的时候，检查当前节点与下一点是否为相同值，如果相同，继续查找祥同值的最大长度，然后指针改变指向。

3、解题代码

3.1 暴力遍历

package pers.klb.jzoffer.hard;

import java.util.HashSet;

/**
 * @program: JzOffer2021
 * @description: 删除链表中重复的结点
 * @author: Meumax
 * @create: 2020-07-20 11:48
 **/
public class DeleteDuplication {
    public ListNode deleteDuplication(ListNode pHead) {
        if (pHead == null) return null;
        HashSet<Integer> set = new HashSet<>();
        ListNode index = pHead;

        // 重复值全部用 set 保存起来
        while (index.next != null) {
            if (index.val == index.next.val) {
                set.add(index.val);
            }
            index = index.next;
        }

        ListNode root = pHead;  // 待返回的新根结点
        // 当前面的结点就是重复的情况下，不断删除，直到不是重复
        while (root != null && set.contains(root.val)) {
            root = root.next;
        }

        ListNode pre = new ListNode(-1); // 上一个结点
        ListNode cur = root;   // 当前结点

        while (cur != null) {
            if (set.contains(cur.val)) {
                if (cur.next == null) {
                    pre.next = null;
                    break;
                } else {
                    cur = cur.next;
                    pre.next = cur;
                    continue;
                }
            }
            pre = cur;
            cur = cur.next;
        }

        return root;
    }
}

class ListNode {
    public int val;
    public ListNode next = null;

    public ListNode(int val) {
        this.val = val;
    }
}

  时间复杂度：O(2N),遍历了2次单链表
  空间复杂度：最坏O(N), 最好O(1)，即最坏全是重复的，最好是一个都没重复

3.2 直接删除

package pers.klb.jzoffer.hard;

/**
 * @program: JzOffer2021
 * @description: 删除链表中重复的结点
 * @author: Meumax
 * @create: 2020-07-20 11:48
 **/
public class DeleteDuplication {
    public ListNode deleteDuplication(ListNode pHead) {
        if (pHead == null) return null;
        ListNode root = new ListNode(-1);
        ListNode pre = root;
        ListNode cur = pHead;
        pre.next = cur;

        while (cur != null) {
            // 当前元素的值和下一个元素的值相同
            if (cur.next != null && cur.val == cur.next.val) {
                cur = cur.next;
                // cur 继续往下走
                while (cur.next != null && cur.val == cur.next.val) {
                    cur = cur.next;
                }
                // 走出上面的循环的时候，此时的 cur 应该是重复元素的最后一个
                cur = cur.next;
                pre.next = cur;
            } else {
                pre = cur;
                cur = cur.next;
            }
        }
        return root.next;
    }
}

class ListNode {
    public int val;
    public ListNode next = null;

    public ListNode(int val) {
        this.val = val;
    }
}

  时间复杂度：O(N)，只遍历了一次链表
  空间复杂度：O(1)

4、解题心得

  链表的遍历也是题型的一种，对链表的插入或删除操作，往往需要注意头结点的问题，一般来说，我们都会定义一个临时头结点，然后真正的头结点是这个临时头结点的 next，目的是防止头结点失去引用。