88. 合并两个有效数组 (Merge Sorted Array)

原题链接

88. 合并两个有效数组 (Merge Sorted Array)

题目大意

You are given two integer arrays nums1 and nums2, sorted in non-decreasing order, and two integers m and n, representing the number of elements in nums1 and nums2 respectively.

Merge nums1 and nums2 into a single array sorted in non-decreasing order.

The final sorted array should not be returned by the function, but instead be stored inside the array nums1. To accommodate this, nums1 has a length of m + n, where the first m elements denote the elements that should be merged, and the last n elements are set to 0 and should be ignored. nums2 has a length of n.

中文释义
给你两个按 非递减顺序 排列的整数数组 nums1nums2,另有两个整数 mn ,分别表示 nums1nums2 中的元素数目。

请你 合并 nums2nums1 中,使合并后的数组同样按 非递减顺序 排列。

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

解题思路:

  1. 定义 index1 指向 nums1 数组最后一个有效数字,index2 指向 nums2 数组最后一个有效数字。
  2. 定义 index 指向 nums1 数组最后一个位置,待填元素。
  3. 依次比较 index1 和 index2 元素大小,较大元素 填入 index 位置,同时相应的指针向前移动。

代码:

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int index1 = m - 1, index2 = n - 1, index = m + n - 1;
        while (index >= 0) {
            if (index1 < 0) nums1[index--] = nums2[index2--];
            else if (index2 < 0) nums1[index--] = nums1[index1--];
            else if (index1 >= 0 && index2 >= 0) {
                if (nums1[index1] >= nums2[index2]) {
                    nums1[index--] = nums1[index1--];
                } else {
                    nums1[index--] = nums2[index2--];
                }
            }
        }
    }
};

你可能感兴趣的:(LeetCode,面试经典150题,算法)