Day_19 —— L e e t C o d e 88 LeetCode 88 LeetCode88:合并两个有序数组
由于两个数组都是有序的,所有只要按顺序比较大小即可。题目中说了 nums1 数组有足够大的空间,又给了m和n,那就知道了混合之后的数组的大小,这样就从 nums1 和 nums2 数组的末尾开始一个一个比较,把较大的数,按顺序从后往前加入混合之后的数组末尾。
设置三个变量 i,j,k,分别指向 nums1,nums2,和混合数组的末尾。进行 while 循环,如果i和j都大于0,再看如果 nums1[i] > nums2[j],说明要先把 nums1[i] 加入混合数组的末尾,加入后k和i都要自减1;反之就把 nums2[j] 加入混合数组的末尾,加入后k和j都要自减1。循环结束后,有可能i或者j还大于等于0,若j大于0,那么还需要继续循环,将 nums2 中的数字继续拷入 nums1。若是i大于等于0,那么就不用管,因为混合数组本身就放在 nums1 中。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i = m-1;
int j = n-1;
int k = m+n-1;
while (i>=0 && j>=0){
if (nums1[i] > nums2[j])
nums1[k--] = nums1[i--];
else
nums1[k--] = nums2[j--];
}
while (j>=0){
nums1[k--] = nums2[j--];
}
}
};
我们还可以写的更简洁一些,将两个 while 循环融合到一起,只要加上 i>=0 且 nums1[i] > nums2[j] 的判断条件,就可以从 nums1 中取数,否则就一直从 nums2 中取数。
class Solution{
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n){
int i = m-1;
int j = n-1;
int k = m+n-1;
while(j >= 0){
nums1[k--] = i >= 0 && nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
}
}
};
线下完整代码实现:
#include <iostream>
#include <vector>
using namespace std;
class Solution
{
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
int i = m-1;
int j = n-1;
int k = m+n-1;
while(j >= 0){
nums1[k--] = i >= 0 && nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
}
}
};
int main()
{
vector<int> nums1 = {1,2,3,0,0,0};
vector<int> nums2 = {2,5,6};
int m = 3;
int n = 3;
Solution().merge(nums1, m, nums2, n);
for (auto& num : nums1)
{
cout << num << " ";
}
cout << endl;
return 0;
}