移除元素(简单)
1. 题目描述
- 给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
- 不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
- 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 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。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
2. 个人解题思路
public static int removeElement(int[] nums, int val){
// 初始化指针
int i = nums.length-1;
// 遍历数组
for (int j = nums.length - 1; j >= 0 ; j--) {
// 判断是否等于val
if (nums[j] == val){
// 交换位置
nums[i] = nums[i] + nums[j] - (nums[j] = nums[i]);
// 指针前移
i--;
}
}
return i+1;
}
注:定义了一个指针指向数组中末尾元素,从后往前遍历该数组,判断有与目标符合的元素则与指针索引位置元素交换位置,指针前移一位,该方法可以保证数组数据的完整性
3. 官方解题参考
public static int removeElement(int[] nums, int val){
// 定义指针
int ans = 0;
// 遍历数组
for(int num: nums) {
// 判断是否不等于val
if(num != val) {
// 不等于则后面的元素替换到前面的位置中,等于则等待被替换,替换后指针后移一位
nums[ans++] = num;
}
}
// 返回有效长度
return ans;
}
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum