leetcode2(移除元素)
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素。元素的顺序可能发生改变。然后返回 nums
中与 val
不同的元素的数量。
假设 nums
中不等于 val
的元素数量为 k
,要通过此题,您需要执行以下操作:
- 更改
nums
数组,使nums
的前k
个元素包含不等于val
的元素。nums
的其余元素和nums
的大小并不重要。 - 返回
k
。
用户评测:
评测机将使用以下代码测试您的解决方案:
int[] nums = [...]; // 输入数组 int val = ...; // 要移除的值 int[] expectedNums = [...]; // 长度正确的预期答案。// 它以不等于 val 的值排序。int k = removeElement(nums, val); // 调用你的实现assert k == expectedNums.length; sort(nums, 0, k); // 排序 nums 的前 k 个元素 for (int i = 0; i < actualLength; i++) {assert nums[i] == expectedNums[i]; }
如果所有的断言都通过,你的解决方案将会 通过。
示例 1:
输入:nums = [3,2,2,3], val = 3 输出:2, nums = [2,2,_,_] 解释:你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。 你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。
注意:
1.移除元素适合用列表操作,removeIf,用数组直接操作,比较复杂且容易超时
2.注意返回的数组长和边界
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
class Solution {public int removeElement(int[] nums, int val) {int k=0;int len=0;List<Integer> list = new ArrayList<>();for (int num : nums) {list.add(num);}// 删除所有目标元素list.removeIf(n -> n == val);list.removeIf(n->n==-1);// Java 8+的lambda表达式// 转回数组for (int i = 0; i < list.size(); i++) {nums[i] = list.get(i);}return list.size();}public static void main(String[] args) {Scanner in = new Scanner(System.in);int[] n= new int[5];for (int i = 0; i < n.length; i++) {n[i]=-1;}int nline=0;String stringin = in.nextLine().trim();if(!stringin.isEmpty()){String[] str = stringin.split(",");for(String str1 : str){n[nline]=Integer.parseInt(str1.trim());nline++;}}int val=in.nextInt();Solution solution = new Solution();solution.removeElement(n,val);}
}