LeetCode27移除元素
题目
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val的
元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅
使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需
要考虑数组中超出新长度后面的元素。
分析
1、这是一道LeetCode的简单题,其核心的思想就是删除数组元素,并且将未被删除的元素前移,我们大多数人的想法都是,如果数组元素等于想要删除的元素,那么就让当前元素等于后面的元素
2、但是这么想就有点麻烦,如果咱们逆着想,如果数组元素不等于要删除的元素,那么就将这个元素存入一个新的数组,那么问题就迎刃而解了。那么j的数值就是移除后数组的新长度
来个完整代码
import java.util.Scanner;
public class Practice1 {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int n =scan.nextInt();
int []arr = new int[n];
for (int i = 0; i < arr.length; i++) {
arr[i] = scan.nextInt();
}
System.out.println("请输入要删除的元素");
int num = scan.nextInt();
int j =0;
for (int i = 0; i < arr.length; i++) {
if(arr[i] != num){
arr[j] = arr[i];
j++;
}
}
System.out.println(j);
}
}
LeetCode答案
class Solution {
public int removeElement(int[] nums, int val) {
int j = 0;
for (int i = 0; i < nums.length; i++) {
if(nums[i] != val){
nums[j]=nums[i];
j++;
}
}
return j;
}
}