【LintCode题集】Q6、Q64
最近开始刷LintCode上的题目,先从标签为容易的开始刷。今天刷的这两题目差不多为同一类型的题目,都是为按照一定的规则合并两个已经有序的数组。
[Q6]
描述:
合并两个排序的整数数组A和B变成一个新的数组。
样例:
给出A=[1,2,3,4],B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]
思路:
题目的方法原型为:
1 public int[] mergeSortedArray(int[] A, int[] B) {}
所以即从A和B数组中一次取数比较大小,将其中较小者放入新的数组C。
当其中一个数组已经全部取出,则将剩余一个数组中的数全部放入数组C中。
在归并排序中,当把一个序列分解为若干个子序列之后,正是通过递归的合并子序列来完成排序。
实现:
1 public int[] mergeSortedArray(int[] A, int[] B) { 2 // Write your code here 3 4 int[] C = new int[A.length + B.length]; 5 6 int indexa = 0; //表示a数组的索引位置 7 int indexb = 0; //表示b数组的索引位置 8 int indexc = 0; //表示c数组的索引位置 9 10 //当某个数组已经全部合并进新的数组的时候跳出循环 11 while( ! ((indexa == A.length) || (indexb == B.length))){ 12 if(A[indexa] <= B[indexb]){ 13 C[indexc ++] = A[indexa ++]; 14 } 15 else{ 16 C[indexc ++] = B[indexb ++]; 17 18 } 19 } 20 //A数组还有剩余元素 21 while(indexa < A.length){ 22 C[indexc ++] = A[indexa ++]; 23 } 24 //B数组还有剩余元素 25 while(indexb < B.length){ 26 C[indexc ++] = B[indexb ++]; 27 } 28 29 return C; 30 31 }
[Q64]
描述:
合并两个排序的整数数组A和B变成一个新的数组。
你可以假设A具有足够的空间(A数组的大小大于或等于m+n)去添加B中的元素。
给出 A = [1, 2, 3, empty, empty]
, B = [4, 5]
合并之后 A 将变成 [1,2,3,4,5]
思路:
题目的方法原型为:
1 public void mergeSortedArray(int[] A, int m, int[] B, int n) {}
从题目函数原型和题目描述可以看出,这次在合并的时候是在A数组的基础上进行合并,A数组的大小大于或等于A+B的大小,所以可以将B数组合并到A数组中。
考虑两种情况:
1 当B中的某个元素已经比A中的最后一个元素还要大时,此时不需要再继续比较了,直接将B中包括当前元素的剩余元素全部插入到A数组最后。
2 当B中的第x元素通过比较插入到了A数组的y位置上,则第x + 1 元素直接从y + 1 位置开始继续比较,前面的无需比较。
此题目比上面的题目还要多增加一个步骤,因为是在A数组上进行插入,所以插入之后,插入位置以后的元素需要往后移动一个位置,涉及到一个移动操作。
实现:
1 public void mergeSortedArray(int[] A, int m, int[] B, int n) { 2 int indexa = 0 ; // 标识A数组的遍历位置 3 int indexb = 0 ; //标识B数组的遍历位置 4 int lastIndex = m - 1 ; //标识A数组中最后的位置 5 6 while ( (indexa <= lastIndex) && (indexb <= n - 1) ) { 7 if(B[indexb] < A[indexa]){ 8 //这个题目第一此提交出现了错误, 错误原因是下面的for循环中应该是i -- , 手滑写成了 i ++ 9 for(int i = lastIndex ; i >= indexa ; i --){ 10 A[i + 1] = A [i] ; 11 } 12 A[indexa ++] = B[indexb ++]; 13 lastIndex ++ ; 14 } 15 else{ 16 indexa ++ ; 17 } 18 } 19 while(indexb <= n - 1 ) { 20 A[++ lastIndex] = B[indexb ++] ; 21 } 22 }
总结:
有关于数组的操作,大多可以用过设置不同变量,来标识数组中的不同位置,从而可以进行不同的操作。