贪心算法的简单题就先这样略过吧,简单题太简单,以至于很难看出贪心的成分。
从今天开始,开始刷《数组类算法》这个专题。
由于这个专题的一些题目以前已经做过了, 所以很多题就不重复做了,第一道题就是比较难的了,啊哈。
一. 题目描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定 nums = [1,1,1,2,2,3],
函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为
1, 1, 2, 2, 3 。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,0,1,1,1,1,2,3,3],
函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为
0, 0, 1, 1, 2, 3, 3 。
你不需要考虑数组中超出新长度后面的元素。
二. 思路
这道题的解法我想到的是两种。
其中一种是通过不断地将不正确的元素与后面的元素交换位置,最后另后续的元素都往前移动一位,不正确元素移动到末尾为止,最后只要返回正确的长度即可,这种方法非常耗时,时间复杂度是O(n^2)。明显不是我们想要的方法。
另一种方法是,将这个初始数组当前两个数组来看待,一个是初始数组本身,一个则是空的数组,也就是说,这个数组在实际操作中,被认为是有两个存在方式的,而实现此方法的关键就是设置两个下标,一个下标用于初始数组,一个小标用于空数组。
三. 代码
class Solution {
public int removeDuplicates(int[] nums) {
int pre = 0;
if(nums.length <= 2)
return nums.length;
for(int i = 2; i < nums.length; i++)
{
if( nums[i] != nums[pre] )
{
nums[pre+2] = nums[i];
pre += 1;
}
}
return pre + 2;
}
}
四. 引用
LeetCode: https://leetcode-cn.com/explore/featured/card/all-about-array/230/define-with-good-care/955/