第一题:移除元素
题目要求: 给一个数组nums 和一个值val ,你需要 原地 移除所有所有数值等于val的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用0(1) 的额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑超出新长度后面的元素。
解决思路:
我们可以考虑使用双指针,dst和src,遍历整个数组。
-
当src指向的元素不等于val就把这个元素赋值给dst指向的位置,dst和src各自指向向下一个位置
-
当src指向的元素等于val就让src独自指向下一个位置。
-
遍历结束条件是src不等于数组有效元素的个数,如果等于就说明没有数据可以判断了。
有了以上思路我们就可以实现代码:
cpp
int removeElement(int* nums, int numSize, int val)
{
int dst = 0, src = 0;
while (src < numSize)
{
if (nums[src] == val)
{
src++;
}
else
{
nums[dst++] = nums[src++];
}
}
return dst; //返回有效元素个数
}
第二题:合并两个有序数组
题目要求: 给你两个按 非递减顺序 排列的整数数组nums1和 nums2 ,另有两个整数m和n,分别表示nums1和nums2的元素数目。
请你合并 nums2到nums1中,使合并后的数组同样按 非递减顺序 排序。
注: 最终,合并后数组不应由函数返回,而是存储在数组nums1 中。为了应对这种情况,nums1 的初始长度为m + n ,其中前m 个元素表示应合并的元素,后n 个元素为
0
,应忽略。nums2 的长度为 n。
解决思路:
既然两个数组都是有序序列,我们就可以从后面开始,判断哪个元素大,就把该元素放在nums1的最后,已知我们需要定义s1、s2、s3,分别定义nums1的有效元素个数、nums2的有效元素个数和nums1的后面空余位置。如果s1和s2位置的元素比大小,谁大就将谁的元素放在s3位置,然后就自减一个位置。
有了以上思路我们就可以实现代码:
cpp
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n)
{
int s1 = m - 1, s2 = n - 1;
int s3 = m + n - 1;
while (s1 >= 0 && s2 >= 0)
{
if (nums1[s1] > nums2[s2])
{
nums1[s3--] = nums1[s1--];
}else
{
nums1[s3--] = nums2[s2--];
}
}
while (s2 >= 0)
{
nums1[s3--] = nums2[s2--];
}
}