原题:
假设有两个有序的整型数组int *a1, int *a2,长度分别为m和n.试用C语言写出一个函数选取两个数组中最大的K个值(K可能大于m+n)写到int *a3中,保持a3降序,并返回a3实际的长度。
函数原型为int merge(int *a3, int *a1, int m, int *a2, int n, int k)
解题思路:此题为两个有序数组的合并:
设置两个下标索引 i和j,逐个比较a1[i]和a2[j],大的进入a3;
当a1或者a2已经全部被排序,就将另一个数组部分拷贝到a3.
- #include <stdio.h>
- int merge(int *a3, int *a1, int m, int *a2, int n, int k)
- {
- int i, j, t;
- if(k > m+n)
- k = m+n;
- for(i = 0, j = 0, t = 0; t < k; t++)
- {
- if(i == m) //若a1片段已经全部被排序
- {
- while(j <= k-m)
- a3[t++] = a2[j++];
- break;
- }
- else if(j == n) //若a2片段已经全部被排序
- {
- while(i <= k-n)
- a3[t++] = a2[i++];
- break;
- }
- if(a1[i] > a2[j])
- {
- a3[t] = a1[i];
- i++;
- }
- else if(a1[i] <= a2[j])
- {
- a3[t] = a2[j];
- j++;
- }
- }
- return k;
- }
- int main()
- {
- int a1[7] = {19,14,13,12,9,6,5};
- int a2[9] = {100,56,34,16,10,7,6,3,1};
- int a3[20];
- int len, k, i;
- for(i = 0; i<7; i++)
- printf(" %d ", a1[i]);
- printf("/n");
- for(i = 0; i<9; i++)
- printf(" %d ", a2[i]);
- printf("/n");
- printf("insert k =");
- scanf("%d", &k);
- len = merge(&a3, &a1, 7, &a2, 9, k);
- for(i = 0; i < len; i++)
- printf(" %d ", a3[i]);
- printf("/n");
- printf("len = %d/n", len);
- system("pause");
- }