前言
提示:这里可以添加本文要记录的大概内容:
删除有序数组中的重复项 II
—LeetCode题目地址:https://leetcode.cn/problems/remove-duplicates-from-sorted-array-ii/description//
提示:以下是本篇文章正文内容,下面案例可供参考
一、题目
删除有序数组中的重复项 II
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例 1:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3。 不需要考虑数组中超出新长度后面的元素。
示例 2:
输入:nums = [0,0,1,1,1,1,2,3,3]
输出:7, nums = [0,0,1,1,2,3,3]
解释:函数应返回新长度 length = 7, 并且原数组的前七个元素被修改为 0, 0, 1, 1, 2, 3, 3。不需要考虑数组中超出新长度后面的元素。
提示:
1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按升序排列
二、思路
第一种思路:
是不满足题目的空间复杂度要求的,但是我只会这种笨一点的方法。
就是把数组的元素以及他有多少个放在treemap里面,key:num[i]、value:n,这里不要用hashmap因为它是不保证元素有序的。然后去遍历先用map.keySet()得到所有的键,然后遍历,判断如果元素个数大于等于2则在原数组中连续保存两次,如果只有一个,则保存一次,每次保存后数组下标后移,最后数组下标i就是返回的数组长度。
第二种思路:
这个是看题解的,用的是快慢指针。
二、代码实现
第一种方法代码实现
class Solution {
public int removeDuplicates(int[] nums) {
if(nums.length==1)
return 1;
Map <Integer,Integer> map=new TreeMap();
for(int i=0;i<nums.length;i++){
if(map.containsKey(nums[i]))
{
map.put(nums[i],map.get(nums[i])+1);
}
else{
map.put(nums[i],1);
}
}
Set<Integer> set=map.keySet();
int i=0;
for(int k:set)
{
if(map.get(k)>=2)
{
nums[i]=k;
i++;
nums[i]=k;
i++;
}
else{
nums[i]=k;
i++;
}
}
return i;
}
}
第二种方法参考leetcode题解。
总结
提示:感觉用多了集合,总是会想着用集合解决问题,但是对于一些有要求的题目,虽然能够通过,但是性能上很难满足要求。