leetcode 双指针 移除元素 java

题目描述
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。

思路:
双指针:i是慢指针,j是快指针,当nums[j]与给定的值相等时,j加一,继续循环,只要nums[j]和val不相等,递增两个索引,复制nums[j]到nums[i]
【可能数组只有一个元素,该元素与val相等,新数组长度为0,因此长度初始化为0,即与i等值,返回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){
    
    
               nums[i] = nums[j];
               i++;             
           }
       }
       return i;
    }
}

猜你喜欢

转载自blog.csdn.net/stonney/article/details/112140300