【数据结构与算法】双指针思想——数组首部或尾部同向遍历
删除排序数组中的重复项
LeetCode:删除排序数组中的重复项
题目描述:
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例:
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
你不需要考虑数组中超出新长度后面的元素。
思想:
直接修改原数组的值使之变成新数组;用pos来标识新数组末端的位置,由于i的位置永远领先于pos,所以循环过程中不会产生冲突;即在已访问过的位置修改值,可保证额外空间为O(1)。
代码:
c++
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size()==0){
return 0;
}
int pos=0;
for(int i=0;i<nums.size()-1;i++){
if(nums[i]!=nums[i+1]){
nums[++pos]=nums[i+1];
}
}
return ++pos;
}
};
移除元素
LeetCode:移除元素
题目描述:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
思想:
pos标记新数组末端位置,i标记遍历旧数组的位置,i永远领先pos,所以可以直接在旧数组上修改。
代码:
c++
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
// int j=0;
// for(int i=0;i<nums.size()-j;i++){
// if(nums[i]==val){
// nums[i]=nums[nums.size()-j-1];
// j++;
// i--;
// }
// }
// return nums.size()-j;
int j=0;
for(int i=0;i<nums.size();i++){
if(nums[i]!=val){
nums[j]=nums[i];
++j;
}
}
return j;
}
};
合并两个有序数组
LeetCode:合并两个有序数组
题目描述:
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
- 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
- 你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
思想:
从nums1数组的末端向前遍历,这样给数组赋新值时不会与旧的数值产生冲突
代码:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
for(int i=m+n-1;i>=0;--i){
if(n==0){
break;
}
if(m==0||nums1[m-1]<=nums2[n-1]){
nums1[i]=nums2[--n];
}else if(nums1[m-1]>nums2[n-1]){
nums1[i]=nums1[--m];
}
}
}
}