高效合并两个有序数组(Merge Sorted Array)

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.The number of elements initialized in nums1and nums2 are m and n respectively.

问题:
两个有序数组,合并成一个有序数组,假设第一个数组空间足够容纳两个数组。
分析:
考虑到num1数组很大,可以直接在num1数组上进行合并,但是要讲究效率。
如果单纯从前往后合并,那么效率会非常低,因为a数组后面的数字需要不停的移动。

换一种思路,我们采用从后往前合并,首先计算出总长度,设置一个指针从num1数组最后往前移动。

void merge(int* nums1, int m, int* nums2, int n) {
    int newIndex=m+n-1;
    int nums1I=m-1,nums2I=n-1;
    while(nums1I>=0 && nums2I>=0){
        if(nums1[nums1I]>=nums2[nums2I]){
            nums1[newIndex]=nums1[nums1I];
            newIndex--;
            nums1I--;
        }else{
            nums1[newIndex]=nums2[nums2I];
            newIndex--;
            nums2I--;
        }
    }
    while(nums1I>=0){
            nums1[newIndex]=nums1[nums1I];
            newIndex--;
            nums1I--;
    }
    while(nums2I>=0){
            nums1[newIndex]=nums2[nums2I];
            newIndex--;
            nums2I--;
    }
}

总结:字符串合并时,我们更应该考虑从后往前这种思路。

附C++代码

class Solution {
public:
	void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
		vector<int> result;
		vector<int>::iterator iter1=nums1.begin();
		vector<int>::iterator iter2=nums2.begin();
		while(iter1!=nums1.begin()+m && iter2!=nums2.begin()+n){
			if(*iter1<=*iter2){
				result.push_back(*iter1);
				++iter1;
			}else{
				result.push_back(*iter2);
				++iter2;
			}
		}
		while(iter1!=nums1.begin()+m){
			result.push_back(*iter1);
			++iter1;
		}
		while(iter2!=nums2.begin()+n){
			result.push_back(*iter2);
			++iter2;
		}
		nums1.swap(result);
	}
};


郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。