leetcode Merge Sorted Array合并两排序数组

2014-11-24 07:27:10 · 作者: · 浏览: 0

Merge Sorted Array


Given two sorted integer arrays A and B, merge B into A as one sorted array.

Note:
You may assume that A has enough space to hold additional elements from B. The number of elements initialized in A and B are m and n respectively.

这是一个逆思维,不是从前往后填写,而是从后往前填写,这样就可以避免不断的移动数据,只要一次遍历就可以了。

而且不需要使用额外的O(n+m)空间。这里应该是使用了O(m)空间吧,m代表B的长度。

严慧敏的数据结构书也有类似算法,需要O(n+m)的空间复杂度。

编程之美也有类似的题目,也差不多是这里这样的解法。

void merge2(int A[], int m, int B[], int n) 
	{
		for (int i = m+n-1; i >= 0; i--)
		{
			if (m>0 && n>0)
			{
				if (A[m-1] < B[n-1]) A[i] = B[--n];
				else A[i] = A[--m];
			}
			else if (m<=0 && n>0)
			{
				A[i] = B[--n];
			}
			else if (m>0 && n<=0)
			{
				A[i] = A[--m];
			}
		}
	}


参考一下leetcode上的程序,发现,原来可以简洁成这样,极品:

	void merge(int A[], int m, int B[], int n) 
	{
		while (n>0)
		{
			if (m<=0 || A[m-1] < B[n-1]) A[n+m-1] = B[--n];
			else A[n+m-1] = A[--m];
		}
	}