数据结构—数组练习题

限定时间复杂度和空间复杂度的题

练习：原地移除数组内指定值元素

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

画图理解
判断i所指向的元素是否是需要移除的值，是则i往后挪一步再判断，不是则往j指向的位置放，放完后i++,j++

int removeElement(int* nums, int numsSize, int val) {
    int i = 0;
    int j = 0;//双下标i，j控制同一个数组nums，i控制原数组，j控制移除元素后的数组
    while (i < numsSize)//数组开始自我赋值
    {
        if (nums[i] != val)//当i遇到不需要移除的元素
        {
            nums[j] = nums[i];//就赋值给nums[j]
            i++;
            j++;//赋值完后++进入下一次赋值
        }
        else//当i遇到需要移除的元素时，不用赋值
        {
            i++;//直接++跳过此元素，而j则不需要动
        }
    }
    return j ;//最后j就是移除元素后的数组长度，因为i和j最后还++了一次跳到循环的判断部分，跳出了循环
}

练习：有序数组的元素去重

给你一个 非严格递增排列 的数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。

考虑 nums 的唯一元素的数量为 k ，你需要做以下事情确保你的题解可以被通过：

• 更改数组 nums ，使 nums 的前 k 个元素包含唯一元素，并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums 的大小不重要。
• 返回 k 

画图理解：

prev指向的元素和cur指向的元素相同就都往后挪一步prev++,cur++，不相同则把prev指向的元素放到dst指向的位置，放完后dst++,prev++,cur++,循环此过程，直到prev指向最后一个元素

出循环后无论最后2个元素是否相同，都还得把最后一个元素放到dst位置

int removeDuplicates(int* nums, int numsSize) {
    int dst = 0;//用dst指向去重后的数组
    int prev = 0;
    int cur = 1;//prev和cur一前一后指向去重前的数组
    while(prev < numsSize - 1)//直到prev指向最后一个元素，则跳出循环
    {
        if(nums[prev] != nums[cur])//如果prev和cur指向的元素不同
        {
            nums[dst] = nums[prev];//则把prev指向的元素放到dst指向的位置
            prev++;
            cur++;
            dst++;//赋值后，3个下标都往后挪一步
        }
        else//否则prev和cur指向的元素相同，则不往dst放，dst保持不动
        {
            prev++;
            cur++;//prev和cur往后挪一步
        }
    }
    nums[dst] = nums[prev];//跳出循环后，再将最后一个元素放到dst指向的位置
    dst++;//dst往后挪一步就是去重后数组元素的个数
    return dst;//返回去重后数组元素个数即可
}

 练习：合并2个有序数组

方法1：时间复杂度O(m+n),空间复杂度也是O(m+n)

dst直接指向nums1，进行原地操作，若从左往右、从小到大操作，会存在nums1中还没放入dst目标位置的元素被覆盖的问题，不信可以画图分析

给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

画图分析

到此i已经=m，跳出循环，nums2还有部分元素5和6未放入tmp空间，放入即可

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n) {
    int* tmp = (int*)malloc(sizeof(int)*(m+n));//申请临时空间存合并后的数组
    int dst = 0;//dst作为临时空间的下标
    int i = 0;//i作为数组nums1的下标
    int j = 0;//j作为数组nums2的下标
    while(i < m && j < n)//开始往临时空间放元素直到nums1或nums2有效数据被遍历完
    {
        if(nums1[i] <= nums2[j])//比较2个数组的元素，较小的放到临时空间
        {
            tmp[dst] = nums1[i];
            i++;
            dst++;
        }
        else
        {
            tmp[dst] = nums2[j];
            j++;
            dst++;
        }
    }//循环结束
    while(j < n)//若j<n，则说明nums2还没遍历完，还有元素未放入临时空间
    {
        tmp[dst] = nums2[j];//把剩余元素全部放入临时空间
        dst++;
        j++;
    }
    while(i < m)//若i<m,则说明nums1还有元素未放入临时空间
    {
        tmp[dst] = nums1[i];//把剩余元素全部放入临时空间
        dst++;
        i++;
    }
   memcpy(nums1,tmp,sizeof(int)*(m+n));//将临时空间的数据拷贝至nums1
    free(tmp);//释放临时空间
    tmp = NULL;
}

优化方法1：时间复杂度O(m+n)，空间复杂度O(1)

从后往前、从大往小放入dst指向的位置

画图分析

void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n) {
    //优化后时间复杂度是O(m+n),空间复杂度是O(1)
    //直接原地从后我往前放，就不会存在后面还没放进dst位置的值被覆盖的问题了
    int dst = m + n -1;//dst最开始指向nums1的容量末端
    int i = m -1;//i最开始指向nums1的有效元素末端
    int j = n -1;//j最开始指向nums2的有效元素末端
    while(i >= 0 && j >= 0)
    {
        if(nums2[j] >= nums1[i])//谁大就是往dst指向的位置放
        {
            nums1[dst] = nums2[j];
            j--;//从后往前比
            dst--;//从后往前放
        }
        else
        {
            nums1[dst] = nums1[i];
            i--;
            dst--;
        }
    }
    while(i < 0 && j >= 0)//如果是i指向的nums1先结束,则肯定是nums2还有元素未合并
    {
        nums1[dst] = nums2[j];//把nums2里剩下未合并的元素全部合并到nums1里即可
        j--;
        dst--;
    }
    //如果是nums2先结束，则无需处理
}