leetcode --- 27. 移除元素(数组)

27. 移除元素

给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例1:

nums = [3,2,2,3], val = 3,函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。

示例2:

给定 nums = [0,1,2,2,3,0,4,2], val = 2,函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。 你不需要考虑数组中超出新长度后面的元素。

JAVA

思路

如果是要删除的元素,那么count次数加一;
如果不是所要删除的元素,用当前元素下标减去count次数
(该元素向前移动)

public static int removeElement(int[] nums, int val) {
    int count=0;
    for(int i =0;i<nums.length;i++){
       if(nums[i]==val){
          count++;
        }else{
          nums[i-count]=nums[i];
           }
       }
        return nums.length-count;//删除元素后的数组长度
    }

发布了20 篇原创文章 · 获赞 23 · 访问量 560

猜你喜欢

转载自blog.csdn.net/m0_45097186/article/details/103654525