三道数组经典题
今天为大家分享关于数组的三道非常经典的题目
一.原地删除元素
这里可以直接使用暴力方法,遍历数组,每次找到指定元素后,就将它后面的所有元素向前挪动一位覆盖掉它,也就是顺序表中删除元素的操作。暴力求解的时间复杂度为O(n^2),不满足要求,下面介绍双指针的方法。由于数组的存储空间在物理上是连续的,故可用下标来代替指针。
定义src和dest两个变量作为数组的下标,用src遍历数组,dest用来更新数组。
若nums[src]的值不是val,则将它插入到dest指向的位置,并让dest指向下一个位置,src继续往后遍历;
若nums[src]的值为val,则跳过此元素,继续往后遍历。
int removeElement(int* nums, int numsSize, int val){
int src = 0;
int dest = 0;
while (src < numsSize)
{
if (nums[src] == val)
{
src++;
}
else
{
nums[dest++] = nums[src++];
}
}
return dest;
}
二. 删除重复元素
该题同样使用双指针的方法求解
定义变量dest和src和作为数组的下标,用src遍历数组,dest用来更新数组。
若nums[src]和nums[dest]不相等,则将它插入到dest指向的位置,并让dest指向下一个位置,src继续往后遍历;
若nums[src] =nums[dest],则跳过此元素,继续往后遍历。
int removeDuplicates(int* nums, int numsSize){
int src = 0;
int dest = 0;
while (src < numsSize)
{
//相同
if (nums[src] == nums[dest])
{
src++;
}
//不相同
else
{
dest++;
nums[dest] = nums[src++];
}
}
return dest + 1;
}
注意这里是有序数组,重复的元素是挨在一起的,若数组无序则不能使用这种方法。
三.合并两个有序数组
本题的思路和前两个题类似,需要设置遍历数组的指针(下标)和更新数组的指针(下标),不同的是,这里有两个数组,所以要两个指针一起遍历数组。
定义end1,end2和dest作为数组下标,其中end1遍历nums1,end2遍历nums2,dest用于更新nums1。
比较nums[end1]和nums[end2],将大的元素插入到dest指向的位置,然后让dest指向前一个位置,继续遍历数组。
当其中一个数组遍历完毕时退出循环,若nums2还有元素未遍历,则将他们全部插入到nums1,若nums1还有元素未遍历则不用管
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
int end1 = m - 1;
int end2 = n - 1;
int dest = m + n - 1;
//把大的放到后面
while (end1 >= 0 && end2 >= 0)
{
if (nums1[end1] >= nums2[end2])
{
nums1[dest--] = nums1[end1--];
}
else
{
nums1[dest--] = nums2[end2--];
}
}
//如果第二个数组还有剩余就全放到第一个数组
if (end2 >= 0)
{
while (end2 >= 0)
{
nums1[dest--] = nums2[end2--];
}
}
}
这里有个小问题,为什么数组要从后往前遍历呢?从前往后遍历可不可以呢?不可以!!!
若是从前往后遍历,比较两个数组大小的时候,如果小的那个元素在nums2中,那么nums1中的元素就会更新,nums1前面的某些元素还没有遍历到就已经改变了。
而从后往前为什么就可以呢?仔细看题,nums1的大小是m+n,后面的n个位置是空出来的。想一个极端情况:nums2的所有元素都比nums1大,就算nums2的所有元素都被插入到nums1中,nums1中原来的元素也不会被覆盖。