【题目描述】
给你两个按 非递减顺序 排列的整数数组 nums1
和 nums2
,另有两个整数 m
和 n
,分别表示 nums1
和 nums2
中的元素数目。
请你 合并 nums2
到 nums1
中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1
中。为了应对这种情况,nums1
的初始长度为 m + n
,其中前 m
个元素表示应合并的元素,后 n
个元素为 0
,应忽略。nums2
的长度为 n
。
【测试用例】
示例1:
输入:nums1 = [1,2,3,0,0,0],m = 3,nums2 = [2,5,6],n = 3
输出:[1,2,2,3,5,6]
解释:需要合并 [1,2,3] 和 [2,5,6] 。合并结果是 [1,2,2,3,5,6]
示例2:
输入:nums1 = [1],m = 1,nums2 = [],n = 0
输出:[1]
解释:需要合并 [1] 和 [ ] ,合并结果是 [1]
示例3:
输入:nums1 = [0],m = 0,nums2 = [1],n = 1
输出:[1]
解释:需要合并的数组是 [ ] 和 [1] 。合并结果是 [1] 。注意:因为 m = 0,所以nums1中没有元素。nums1中仅存的0仅仅是为了确保合并结果可以顺利存放到nums1中。
【思路分析】
法一:先合并再排序
先说我的做法,先合并再排序。输入的nums1数组中末位n个元素都是0,是给nums2中的元素预留的位置,那么很自然的就可以想到直接把nums2中的元素放到这n个空位上,这样一来所有的元素就放在了nums1中(无序),再通过一个排序算法(我使用的最简单的冒泡排序)将nums1排序就行了。
法二:三指针合并
三指针合并(不是专门的名词,只是我这么叫)是题解的一种做法,法一的时间复杂度因为有冒泡排序所以是O(n),法二可以降低到O(m+n)。
三个指针变量pos1,pos2,pos分别指向nums1数组有效元素的末尾,nums2数组的末尾,nums1数组的末尾,即pos1=m-1,pos2=n-1,pos3=m+n-1。
当 pos2>=0 时进入循环,在每轮循环中判断 nums1[pos1] 和 nums2[pos2] 的关系,把大的元素放到nums1中的pos位置上:
如果 pos1>=0 && nums1[pos1] > nums2[pos2],则 nums1[pos--] = nums1[pos1--];
否则 nums1[pos--] = nums2[pos2--]。
ps:循环条件是pos2>=0,当退出循环时,pos2<0,此时nums2中的所有元素都放入了nums1中对应的位置(最终位置),此时即使nums1中还有没有移动的元素,这时这些没有移动的元素也一定是在其最终位置上,所以循环条件只有pos2>=0即可。
【参考代码】
法一:C实现
#include <stdio.h>
#include <stdlib.h>
//easy-88-合并两个有序数组
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n);
int main(){
int m=3, n=3;
int nums1[] = {4,5,6,0,0,0};
int nums2[] = {1,2,3};
int nums1Size = sizeof(nums1) / sizeof(nums1[0]);
int nums2Size = sizeof(nums2) / sizeof(nums2[0]);
merge(nums1, nums1Size, m, nums2, nums2Size, n);
int i;
for(i=0;i<nums1Size;i++){
printf("%d ",nums1[i]);
}
return 0;
}
//法一:先合并再排序
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n) {
int i, k=0;
for(i=m;i<nums1Size;i++){
nums1[i] = nums2[k++];
}
int j;
for(i=0;i<nums1Size;i++){
for(j=i+1;j<nums1Size;j++){
if(nums1[i] > nums1[j]){
int tmp = nums1[i];
nums1[i] = nums1[j];
nums1[j] = tmp;
}
}
}
}
法一:C++实现
#include <iostream>
#include <vector>
using namespace std;
//easy-88-合并两个有序数组
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n);
};
//法一:先合并再排序
void Solution::merge(vector<int>& nums1, int m, vector<int>& nums2, int n){
int i, k=0;
for(i=m;i<nums1.size();i++){
nums1[i] = nums2[k++];
}
int j;
for(i=0;i<nums1.size();i++){
for(j=i+1;j<nums1.size();j++){
if(nums1[i] > nums1[j]){
int tmp = nums1[i];
nums1[i] = nums1[j];
nums1[j] = tmp;
}
}
}
}
int main(){
int m=3, n=3;
vector<int> nums1 = {1,2,3,0,0,0};
vector<int> nums2 = {2,5,6};
Solution sol;
sol.merge(nums1, m, nums2, n);
int i;
for(i=0;i<nums1.size();i++){
cout<<nums1[i]<<" ";
}
return 0;
}
法二:C实现
#include <stdio.h>
#include <stdlib.h>
//easy-88-合并两个有序数组
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n);
int main(){
int m=3, n=3;
int nums1[] = {4,5,6,0,0,0};
int nums2[] = {1,2,3};
int nums1Size = sizeof(nums1) / sizeof(nums1[0]);
int nums2Size = sizeof(nums2) / sizeof(nums2[0]);
merge(nums1, nums1Size, m, nums2, nums2Size, n);
int i;
for(i=0;i<nums1Size;i++){
printf("%d ",nums1[i]);
}
return 0;
}
//法二:三指针合并
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n) {
int pos1=m-1, pos2=n-1, pos=nums1Size-1;
while(pos2 >= 0){
if(pos1 >= 0 && nums1[pos1] > nums2[pos2]){
nums1[pos--] = nums1[pos1--];
}else{
nums1[pos--] = nums2[pos2--];
}
}
}
法二:C++实现
#include <iostream>
#include <vector>
using namespace std;
//easy-88-合并两个有序数组
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n);
};
//法二:三指针合并
void Solution::merge(vector<int>& nums1, int m, vector<int>& nums2, int n){
int pos1=m-1, pos2=n-1, pos=nums1.size()-1;
while(pos2 >= 0){
if(pos1 >= 0 && nums1[pos1] > nums2[pos2]){
nums1[pos--] = nums1[pos1--];
}else{
nums1[pos--] = nums2[pos2--];
}
}
}
int main(){
int m=3, n=3;
vector<int> nums1 = {1,2,3,0,0,0};
vector<int> nums2 = {2,5,6};
Solution sol;
sol.merge(nums1, m, nums2, n);
int i;
for(i=0;i<nums1.size();i++){
cout<<nums1[i]<<" ";
}
return 0;
}