LeetCode27—移除元素(java版)

题目描述:

标签:数组   双指针

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

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

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

  代码:

思路分析:想到用双指针了,但是我的双指针移动方法没有官方解答的好,所以这里借鉴一下了。

1、因为题目说“不需要考虑数组中超出新长度后面的元素”,所以这里选择把前面出现的元素移到数组尾部。

2、即定义双指针,第一个指针 i 指向数组头,第二个指针指向数组尾部。

3、循环数组,当第一个指针指向的元素不等于val值时,则向右移动;如果等于val值时,这里用了一个比较巧妙的处理:让第一个指针元素指向的元素与第二个指针指向的元素交换,这时数组的有效长度是减1的,即n-1。这里会再检验移动后的元素,所以不用担心第二个指针指向的元素=val值。我觉得他的巧妙就是在于不断地改变n,不断得到数组清除元素后的有效长度。

class Solution {
    public int removeElement(int[] nums, int val) {
        int i = 0;
        int n = nums.length;
        while(i < n){
          if(nums[i] == val){
              nums[i] = nums[n - 1];
              n--;
          }else{
              i++;
          }
        }
        return n;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_40840749/article/details/114292549