LeetCode 移除重复节点

题目

/**
 * @Author:linjunbo
 * @Description:
 * @Date: 2020/3/17 10:13
 */

/**
 * 编写代码，移除未排序链表中的重复节点。保留最开始出现的节点。
 * 

 * 示例1:
 * 

 * 输入：[1, 2, 3, 3, 2, 1]
 * 输出：[1, 2, 3]
 * 示例2:
 * 

 * 输入：[1, 1, 1, 1, 2]
 * 输出：[1, 2]
 * 提示：
 * 

 * 链表长度在[0, 20000]范围内。
 * 链表元素在[0, 20000]范围内。
 * 

 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/remove-duplicate-node-lcci
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

解题

  public static ListNode removeDuplicateNodes(ListNode head) {
       //解题思路。 使用set集合 判断add添加是否能成功  如果不成功则说明存在重复 将前一个结点的的next指向当前结点的next
       //当前节点
       ListNode thisNode = head;
       //上一个节点
       ListNode beforeNode = null;
       Set<Integer> set = new HashSet<>();
       while (thisNode != null) {
           if (!set.add(thisNode.val)) {

               beforeNode.next = thisNode.next;
               thisNode.next = null;
               thisNode = beforeNode.next;


           } else {
               beforeNode = thisNode;
               thisNode = thisNode.next;

           }
       }

       return head;

   }