LeetCode——移除元素

LeetCode——移除元素

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

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

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

示例 1:
[3,2,2,2]
给定 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。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。

解题思路：
我定义了两个指针，一个指向数组的头部，一个指向尾部，当 i 指向的等于var，j 指向的不等于var时，将 j 指向的数赋值给 i 指向的数，达到消去var的作用。在最后提交了好几次才成功，原因就是没有考虑一些特殊情况，比如说[ 3 , 3 ] , [ 4 , 5 ]之类的特殊情况，总结出原因之后我加了以下这条：

if(nums[i] == var || i > j){//当i最终位置指向的数字等于var时，或者i最终大于j的情况时
            return i;
        }

这是为了防止 i 最终位置指向的数等于var，或者 i > j 的特殊情况。

具体理解可以结合上述的几个特殊例子进行理解

solution代码如下：

package LeetCode.removeElement;

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

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

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

 

示例 1:
[3,2,2,2]
给定 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。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。
 */
public class removeElementSolution {
    public static int solution(int[] nums,int var){
        if(nums.length == 0){//当数组为空时，直接返回0
            return 0;
        }
        int i = 0,j = nums.length - 1;//定义两个数组，一个指向数组头部，一个指向数组尾部
        while(i < j){//当i大于或等于j的时候跳出循环
            if(nums[i] != var){//当i指向的数不等于var时，i向前移动，并开始下一次循环
                i++;
                continue;
            }
            if(nums[j] == var){//当j指向的数等于var时，j向后移动，并开始下一次循环
                j--;
                continue;
            }
            nums[i] = nums[j];//进行到此时，i指向的数等于var，j指向的数不等于var
            i++;
            j--;
        }
        if(nums[i] == var || i > j){//当i最终位置指向的数字等于var时，或者i最终大于j的情况时
            return i;
        }
        return i + 1;
    }
}

测试代码如下：

package LeetCode.removeElement;

import java.util.Scanner;

public class removeElementMain {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        System.out.println("输入数组元素个数：");
        int n = scan.nextInt();
        int nums[] = new int[n];
        System.out.println("输入数组元素：");
        for (int i = 0; i < n; i++) {
            nums[i] = scan.nextInt();
        }
        System.out.println("输入删除元素：");
        int var = scan.nextInt();
        System.out.println("新数组元素个数：" + removeElementSolution.solution(nums,var));
    }
}

测试：


LeetCode测试：