欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

数组-7-LeetCode 27题 -> 移除元素

程序员文章站 2024-03-22 14:03:58
...

LeetCode 27题 -> 移除元素

1. 题目描述
  • 给定一个数组 nums 和一个值 val,需要原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。不需要考虑数组中超出新长度后面的元素。

  • 示例:

    nums = [3,2,2,3], val = 3,
    
    返回新的长度 2, 并且 nums 中的前两个元素均为 2
2. 解题思路
  • 拷贝覆盖!
    • ①遍历数组nums,两个指针i和ans;
    • ②判断nums[i] 是否等于target;
    • ③若相等,跳过nums[i]不进行拷贝覆盖;
    • ④若不相等,则进行拷贝覆盖,令nums[ans] = nums[i],ans+1
    • ⑤最后ans即为新数组的长度。
3. 代码实现
public class RemoveElement {
    public static int removeElement(int[] nums, int target){
        int ans = 0;
        for (int i = 0; i < nums.length; i++) {
            if(nums[i] != target){
                nums[ans] = nums[i];
                ans += 1;
            }
        }
        return ans;
    }

    public static void main(String[] args) {
        int[] nums = {0,1,2,2,3,0,4,2};
        System.out.println(removeElement(nums, 2));
    }
}