[leetcode]Merge Sorted Array

新博文地址:[leetcode]Merge Sorted Array

http://oj.leetcode.com/problems/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 (size that is greater or equal to m + n) to hold additional elements from B. The number of elements initialized in A and B are m and n respectively.

 合并两个排好序的数组A,B,不创建新的数组,即把B数组插入到A数组中,参数m和n分别是数组a和b的初始化的元素个数

 

一般而言,我们如果正向遍历数组不满足时间复杂度的时候,不妨尝试一下逆向遍历。

比如说1,7,9,11和2,3,4,5,8当比较1 和2的时候,不做操作,当比较7和2的时候,将7后移,当比较7和3的时候,再次将7后移,当比较7和4的时候....

我们发现,正向遍历时,某个元素移位后的位置并不一定的最终位置。因此不妨从后向前遍历,两个数组元素一共有9个,那么比较11和8,显然,11是两个数组中最大的,其最终下标是8,再比较9和8,同理,确定9的最终下标,以此类推。

还要注意边界,当m == 0或者n == 0时的相关处理

代码如下:

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

 

你可能感兴趣的:(LeetCode)