题目:有两个数组a[m]和b[n],将它们合并成数组c[m+n],需要c[m+ n]也是有序数组。
两种实现思路:
定义一个新数组,长度为两个数组长度之和,将两个数组都copy到新数组,然后排序。
给两个数组分别定义一个下标,最大长度是数组长度减一,按位循环比较两个数组,较小元素的放入新数组,下标加一(注意,较大元素对应的下标不加一),直到某一个下标超过数组长度时退出循环,此时较短数组已经全部放入新数组,较长数组还有部分剩余,最后将剩下的部分元素放入新数组,大功告成。
- (void)viewDidLoad {
[super viewDidLoad];
//实例化数组 A
NSMutableArray *arrA = [NSMutableArray arrayWithArray:@[@1,@3,@5,@7,@9,@11]];
//实例化数组 B
NSMutableArray *arrB = [NSMutableArray arrayWithArray:@[@8,@15,@17,@20,@22,@35]];
//新数组
NSMutableArray *marr = [NSMutableArray array];
//开始遍历
for(int i = 0; i< 100; i++) {
if (arrA[0] < arrB[0]) {
[marr addObject:arrA[0]];
[arrA removeObject:arrA[0]];
}else{
[marr addObject:arrB[0]];
[arrB removeObject:arrB[0]];
}
NSLog(@"已经循环了--->>%d次",i);
if (arrA.count == 0){
[marr addObjectsFromArray:arrB];
NSLog(@"新数组--->>%@",marr);
return;
}
if (arrB.count == 0){
[marr addObjectsFromArray:arrA];
NSLog(@"新数组---->> %@",marr);
return;
}
}
}
第一种方式应该是大多数人想到的,也比较容易实现,下面主要来实现第二种方式。
代码如下:
i++ :先引用后增加
++i :先增加后引用
//两个有序数组的合并函数
int *MergeList(int a[],int b[])
{
int lena = a.length;
int lenb = b.length;
int *result = (int *)malloc(sizeof(int)*(lena+lenb));
int i=0,j=0,k=0; //i:用于标示a数组 j:用来标示b数组 k:用来标示传入的数组
while(i<lena && j<lenb)
if(a[i] <= b[j]) {
result[k++] = a[i++];
}else{
result[k++] = b[j++];
}
/* 后面连个while循环是用来保证两个数组比较完之后剩下的一个数组里的元素能顺利传入 */
while(i < lena)
result[k++] = a[i++];
while(j < lenb)
result[k++] = b[j++];
return result;
}
}