第一题:移除元素
题目要求:给一个数组nums和一个值val,你需要 原地 移除所有所有数值等于val的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用0(1)的额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑超出新长度后面的元素。
解决思路:
我们可以考虑使用双指针,dst和src,遍历整个数组。
1. 当src指向的元素不等于val就把这个元素赋值给dst指向的位置,dst和src各自指向向下一个位置
2. 当src指向的元素等于val就让src独自指向下一个位置。
3. 遍历结束条件是src不等于数组有效元素的个数,如果等于就说明没有数据可以判断了。
有了以上思路我们就可以实现代码:
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位置,然后就自减一个位置。
有了以上思路我们就可以实现代码:
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--];}
}