数组类算法基础巩固
日期:2022/7/7
做好初始定义
1.移动零
题目描述:给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
输入: nums = [0]
输出: [0]
思路:
遍历数组,将非0的数提前。
用index表示索引,for循环遍历,非0的数提到nums[index]的地方,index++。之后从num[index]开始,将后面的数都变为0
代码+解析:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int index=0;
for(int i=0; i<nums.size(); i++){
if(nums[i]){
nums[index++] = nums[i];
}
}
for(int i=index; i<nums.size(); i++){
nums[i]=0;
}
}
};
用到的技巧:
- 双指针
2.移除元素
题目描述:给你一个数组 nums 和一个值 val,你需要 原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
思路:
遍历数组,将非val的数提前。
和上题差不多,但是不用把后面的数字删除,直接返回index就可以了
代码+解析:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int index=0;
for(int i=0; i<nums.size(); i++){
if(nums[i] != val) nums[index++] = nums[i];
}
return index;
}
};
用到的技巧:依然双指针
3.删除排序数组中的重复项
题目描述:给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。
将最终结果插入 nums 的前 k 个位置后返回 k 。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:
输入:nums = [1,1,2]
输出:2, nums = [1,2,_]
解释:函数应该返回新的长度 2 ,并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出新长度后面的元素。
输入:nums = [0,0,1,1,1,2,2,3,3,4]
输出:5, nums = [0,1,2,3,4]
解释:函数应该返回新的长度 5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
思路:
从1开始遍历数组,如果当前的数不等于前一个数,则替换到nums[index]中,最后返回index
代码+解析:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int index = 1;
for(int i=1; i<nums.size(); i++){
if(nums[i] != nums[i-1]) nums[index++] = nums[i];
}
return index;
}
};
4.删除排序数组中的重复项Ⅱ
题目描述:给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 不需要考虑数组中超出新长度后面的元素。
输入:nums = [0,0,1,1,1,1,2,3,3]
输出:7, nums = [0,0,1,1,2,3,3]
解释:函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。 不需要考虑数组中超出新长度后面的元素。
思路:
nums[i]和nums[index-2]对比,如果不等于,就存入nums[index]中,index++
代码+解析:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int index = 0;
for(int i=0; i<nums.size(); i++){
if(index<2 || nums[i] != nums[index-2] ){ //index<2的话就直接存。一开始我初始化i和index=2,但是会导致size<2的出错,
nums[index++] = nums[i]; //其次,index<2这个条件要放到前面,不然nums[index-2]会出错
}
}
return index;
}
};
**总结:**多运用双指针。其实可以把index看成是对应了一个新数组的索引,然后把nums[i]按照规则存到“新数组中”
运用基础算法思想
1.颜色分类
题目描述:给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
示例:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
输入:nums = [2,0,1]
输出:[0,1,2]
思路:
遇到红色(nums[i]==0)的就放到数组最左边(leftIndex),遇到蓝色(nums[i]==2)的就放到数组最右边(rightIndex),剩下的就是白色(nums[i]==1)
代码+解析:
class Solution {
public:
void sortColors(vector<int>& nums) {
int left=0,right=nums.size()-1;
for(int i=0; i<=right; i++){ //注意:结束条件i<=right,不然2和1会互换
int temp;
if(nums[i]==0){
temp = nums[left];
nums[left++] = nums[i];
nums[i] = temp;
}else if(nums[i]==2){
temp = nums[right];
nums[right--] = nums[i];
nums[i] = temp;
i--; //重新再判断一遍换过去的那个
}
}
}
};
用到的技巧:
- 基数排序法:记录0,1,2的次数,再重新排列。但这样会对数组进行两次遍历
- 三路快速排序方法:设置三个lt,gt,i,定义:nums[0…lt] == 0,nums[lt+1,…,gt-1]=1,nums[gt,…,n-1]=2
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-555LpO7C-1657186379042)(C:\Users\1\AppData\Roaming\Typora\typora-user-images\image-20220707154814159.png)]
2.数组中的第K个最大元素
题目描述:给定整数数组 nums
和整数 k
,请返回数组中第 **k**
个最大的元素。
请注意,你需要找的是数组排序后的第 k
个最大的元素,而不是第 k
个不同的元素。
示例:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
思路:
利用快速排序的思想,从数组 S 中随机找出一个元素 X,把数组分为两部分 Sa 和 Sb。Sa 中的元素大于等于 X,Sb 中元素小于 X。这时有两种情况:
Sa 中元素的个数小于 k,则 Sb 中的第 k-|Sa| 个元素即为第k大数;
Sa 中元素的个数大于等于 k,则返回 Sa 中的第 k 大数。时间复杂度近似为 O(n)
代码+解析:
一开始想到了快排,但是懒得写快排的函数,就偷懒用sort然后下一题。
用到的技巧:快速排序
3.合并两个有序数组
题目描述:给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
示例:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
解释:需要合并 [1,2,3] 和 [2,5,6] 。
合并结果是 [1,2,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
解释:需要合并 [1] 和 [] 。
合并结果是 [1] 。
输入:nums1 = [0], m = 0, nums2 = [1], n = 1
输出:[1]
解释:需要合并的数组是 [] 和 [1] 。
合并结果是 [1] 。
注意,因为 m = 0 ,所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。
思路:
定义一个新数组,长度为m+n,遍历num1和num2,将元素一个个填到新数组中,然后num1=新数组
代码+解析:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int p1 = 0, p2 = 0;
int sorted[m + n];
int cur;
while (p1 < m || p2 < n) {
if (p1 == m) {
cur = nums2[p2++];
} else if (p2 == n) {
cur = nums1[p1++];
} else if (nums1[p1] < nums2[p2]) {
cur = nums1[p1++];
} else {
cur = nums2[p2++];
}
sorted[p1 + p2 - 1] = cur;
}
for (int i = 0; i != m + n; ++i) {
nums1[i] = sorted[i];
}
}
};