LeetCode第[26]题(Java)：Remove Duplicates from Sorted Array 标签：Array

Posted 2020-10-25 清风吹斜阳

题目难度：Easy

题目：

Given a sorted array, remove the duplicates in-place such that each element appear only once and return the new length.

Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.

翻译：

给定一个排好序的数组后，删除重复的元素，这样每个元素只出现一次，并返回新的长度。

不要新建另一个数组分配额外的空间，只能通过修改原有数组，且空间复杂度为O(1)。

示例：[1,1,2]——[1,2]  2

 

思路：一看见消除重复，就想到了Set，然后写了代码如下

1     public int removeDuplicates(int[] nums) {
2         Set s = new HashSet();
3         for (int i = 0; i < nums.length; i++) {
4             s.add(nums[i]);
5         }
6         return s.size();
7     }

但是答案说错误？？

因为此题比较特殊，它不仅仅检查最后的结果，而且检查nums最后的值是否是期望的（只截取最后返回结果的大小）

然后我在后面加了个迭代器循环赋值，结果还是不对？

1         for (Iterator iterator = s.iterator(); iterator.hasNext();) {
2             nums[i++] = (Integer) iterator.next();
3         }

 

顺序也要管？  好吧HashSet无序的，那就用TreeSet吧：

 1     public int removeDuplicates(int[] nums) {
 2         Set<Integer> s = new TreeSet<Integer>();
 3         for (int i = 0; i < nums.length; i++) {
 4             s.add(nums[i]);
 5         }
 6         int i = 0;
 7         for (Iterator iterator = s.iterator(); iterator.hasNext();) {
 8             nums[i++] = (Integer) iterator.next();
 9         }
10         return s.size();
11     }

 161 / 161 test cases passed. Status: Accepted Runtime: 24 ms   beats 5.31%

 

由于此处采用Set占用了额外的空间，空间复杂度为O(N)，虽然结果正确，但是不符合原题意图，下面是参考答案。

 1 public int removeDuplicates(int[] nums) {
 2     if (nums.length == 0) return 0;
 3     int i = 0;
 4     for (int j = 1; j < nums.length; j++) {
 5         if (nums[j] != nums[i]) {
 6             i++;
 7             nums[i] = nums[j];
 8         }
 9     }
10     return i + 1;
11 }

一开始有往这方面想，但是运行后发现边界问题总是处理不了，就放弃了，

此处巧妙地采用了一个循环外的int做指针对原数组进行修改，同时循环内部从第二个开始与指针所指做比较，跳过重复的元素。

 

期间编译错误：

1. if 后面的括号内的判断“==”写成了“=”；

2. Set的toArray方法只能利用传参toArray( new int[set.size()] ) 这样才不会有转型错误，但是这样也只能转为非基本类型（Integer而不能是nt），像转为int数组只能利用迭代器进行循环赋值；

3. 忘记写return。