LeetCode 26
2020.9.24
我的通过代码
class Solution {
public int removeDuplicates(int[] nums) {
int ans = 0;
int current = -10000;
int time = 0;
for(int i=0;i<nums.length;i++) {
if(current==-10000) {
current = nums[i];
time = 1;
ans++; continue;
}
if(nums[i]==current) {
time++;
} else {
current = nums[i];
nums[i-time+1] = nums[i];
ans++;
}
}
return ans;
}
}
这道题本身并不难,但是LeetCode加上了一定的限制条件:
“不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。”
这样一来,事情就有意思的多了。
我的想法是先记录重复元素的个数,在遇到下一个不重复的元素时,将其赋值到之前若干重复元素的首个元素之后,再重置用来记录的变量,进行循环。
这个方法看起来也还行,就是在初始化时,我一开始让current的值为-100,结果我碰到了个从-999开始的用例。
我真是服了。
然后我翻了翻评论,发现大家都在说双指针
class Solution {
public int removeDuplicates(int[] nums) {
int i=0;
for(int j=0;j<nums.length;j++){
if(nums[i]!=nums[j]){
nums[++i]=nums[j];
}
}
return i+1;
}
}
这段代码真是太优雅了,虽然思路和我的想法类似,但是他省去了计数的环节,直接遍历数组而不需另设计数器。
NB!