27、移除元素
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
输入:nums = [3,2,2,3], val = 3 输出:2, nums = [2,2]示例 2:
输入:nums = [0,1,2,2,3,0,4,2], val = 2 输出:5, nums = [0,1,4,0,3]
双指针法:指针i和j,i从前往后遍历数组,j指向最后一个元素的位置,当i位置的值等于要移除的元素时,将其交换到最后面(j所指位置),然后执行i--和j--的操作开始下一次for循环,循环完后j所指位置为移除后的数组最后一个元素的位置,+1返回结果。
class Solution {
public int removeElement(int[] nums, int val) {
int j = nums.length - 1; //最后一个元素的下标
for (int i = 0; i <= j; i++){ //遍历数组
if(nums[i] == val){ //如果等于要被移除的元素
swap(nums, i--, j--); //将其和最后一个位置的元素交换,交换过来的元素还没有看是不是移除的元素,所以i--,最外层的for循环再i++查看交换过来的元素,j也--
}
}
return j+1; //最后一次交换完j--,指的是移除后数组最后一个元素的下标,+1返回数组的长度
}
void swap(int[] nums, int i, int j){ //异或法交换数组中i和j位置的元素
nums[i] = nums[i] ^ nums[j];
nums[j] = nums[i] ^ nums[j];
nums[i] = nums[i] ^ nums[j];
}
}
遍历完后数组分为两部分,【0 ~ j】是移除后的数组,【j ~ nums.length - 1】是要移除的元素。
- 时间复杂度:O(n)
- 空间复杂度:O(1)
283、移动零
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums =[0,1,0,3,12] 输出: [1,3,12,0,0]示例 2:
输入: nums = [0] 输出: [0]
双指针法:两个指针 i 和 j,j记录非零元素的个数,在第一次遍历的时候遇到非零元素就将其往左边挪,遍历完数组后j就指向最后一个非零元素。再遍历一次将j后面的元素置为零。
class Solution {
public void moveZeroes(int[] nums) {
if(nums == null){ //数组为空直接返回
return;
}
int j = 0; //记录非零元素的个数
for(int i = 0; i < nums.length; ++i){
if(nums[i] != 0){ //如果不为零,就将该元素依次放到数组左边
nums[j++] = nums[i];
}
} //for循环结束后j指向最后一个非零元素的下一个位置
for(int i = j; i < nums.length; ++i){ //从该位置开始将后面的值赋值为0
nums[i] = 0;
}
}
}
- 时间复杂度:O(n)
- 空间复杂度:O(1)
485、最大连续1的个数
给定一个二进制数组 nums
, 计算其中最大连续 1
的个数。
示例 1:
输入:nums = [1,1,0,1,1,1] 输出:3示例 2:
输入:nums = [1,0,1,1,0,1] 输出:2
class Solution {
public int findMaxConsecutiveOnes(int[] nums) {
int maxCount = 0, count = 0;
int n = nums.length;
for(int i = 0; i < n; i++){
if(nums[i] == 1){ //如果当前元素是1,则将当前的连续1的个数加1
count++;
}else{
maxCount = Math.max(maxCount, count); //否则,使用之前的连续1的个数更新最大的连续1的个数,并将当前的连续1的个数清零
}
}
maxCount = Math.max(maxCount, count); //遍历数组结束之后,再次使用当前的连续1的个数更新最大的连续1的个数
return maxCount;
}
}
遍历结束后再次使用当前的连续 1 的个数更新最大的连续 1 的个数,因为数组的最后一个元素可能是 1,且最长连续 1 的子数组可能出现在数组的末尾。
-
时间复杂度:O(n)
-
空间复杂度:O(1)