移除元素 Leecode 27. Remove Element

给一个数组和一个值,移除数组中所有该元素并返回新的长度。不可以分配新的空间。

Example 1:

nums = [3,2,2,3], val = 3,
返回长度为2,数组为[2,2]

Example 2:

nums = [0,1,2,2,3,0,4,2], val = 2,
返回长度为5,数组为[0,1,3,0,4]

根据题意,移除数组中值为val的元素后的数组长度,保证算法的空间复杂度是O(1)。
假设数组为nums,首先我们肯定是需要遍历数组的,使用两个指针i和j,指向数组第一个元素。接下来需要判断所遍历到的元素num[j]是否等于val

  • 如果不等于val,i和j都向后移。
  • 如果等于val,j往后移,找到最近的不等于val的元素,让nums[i] = nums[j]。
    在第二种情况下,i的大小就是移除数组中值为val的元素后的数组长度。
    代码如下:
/*第一种情况*/
class Solution {
    public int removeElement(int[] nums, int val) {
        int i = 0;
        for(int j = 0; j < nums.length; j++){
            if(nums[j] != val){//如果不等于val,i和j都向后移。
                i++;
            }
        }
        return i;
    }
}

/*第二种情况*/
class Solution {
    public int removeElement(int[] nums, int val) {
        int i = 0;
        for(int j = 0; j < nums.length; j++){
            if(nums[j] != val){//如果不等于val,i和j都向后移。
                i++;
                nums[i] = nums[j];//此时j为最近的不等于val的元素下标。
            }
        }
        return i;
    }
}

你可能感兴趣的:(LeetCode,Leetcode)