/**
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
-
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/
public class RemoveElement {
/*- 方法一:双指针
- 思路
- 既然问题要求我们就地删除给定值的所有元素,我们就必须用 O(1)的额外空间来处理它。如何解决?我们可以保留两个指针 i 和 j,其中 i 是慢指针,j 是快指针。
- 算法
- 当 nums[j] 与给定的值相等时,递增 j以跳过该元素。只要 nums[j]!=val,我们就复制 nums[j]到 nums[i]并同时递增两个索引。重复这一过程,直到 j 到达数组的末尾,该数组的新长度为 i。
*复杂度分析 - 时间复杂度:O(n), 假设数组总共有 n个元素,i 和 j 至少遍历 2n 步。
- 空间复杂度:O(1)。
- @param nums
- @param val
- @return
*/
public int removeElement(int[] nums, int val) {
if(nums.length==0) return 0;
int i=0;
for(int j=0;j<nums.length;j++){
if(nums[j]!=val){
nums[i]=nums[j];
i++;
}
}
return i;
}
/**
- 方法二:双指针 —— 当要删除的元素很少时
- 思路
- 现在考虑数组包含很少的要删除的元素的情况。例如,num=[1,2,3,5,4],Val=4num=[1,2,3,5,4],Val=4。之前的算法会对前四个元素做不必要的复制操作。另一个例子是 num=[4,1,2,3,5],Val=4num=[4,1,2,3,5],Val=4。似乎没有必要将 [1,2,3,5][1,2,3,5] 这几个元素左移一步,因为问题描述中提到元素的顺序可以更改。
- 算法
- 当我们遇到 nums[i] = val 时,我们可以将当前元素与最后一个元素进行交换,并释放最后一个元素。这实际上使数组的大小减少了 1。
- 请注意,被交换的最后一个元素可能是您想要移除的值。但是不要担心,在下一次迭代中,我们仍然会检查这个元素。
- Java
- 作者:LeetCode
- 链接:https://leetcode-cn.com/problems/two-sum/solution/yi-chu-yuan-su-by-leetcode/
- 来源:力扣(LeetCode)
- 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
- @param nums
- @param val
- @return
*/
public int removeElement2(int[] nums, int val) {
if(nums.length==0) return 0;
int i=0;
int n=nums.length;
while(i<n){
if(nums[i]==val){
nums[i]=nums[n-1];
n--;
}else{
i++;
}
}
return n;
}
public static void main(String[] args){
int[] nums=new int[]{3,2,2,3};
RemoveElement removeElement = new RemoveElement();
int len = removeElement.removeElement2(nums,3);
System.out.println(len);
for (int i = 0; i < len; i++) {
System.out.println(nums[i]);
}
}
}