已知数组A和B各有m、n个元素,且元素按值非递减排列,现要求把A和B归并为一个新的数组C,且C中的数据元素仍然按值非递减排列。
例如,若A=(3,5,8,11),B=(2,6,8,9,11,15,20),
则C=(2,3,5,6,8,8,9,11,11,15,20)
输入格式:
首先输入一个正整数T,表示测试数据的组数,然后是T组测试数据。
每组测试数据输入两行,其中第一行首先输入A的元素个数m(1≤m≤100),然后输入m个元素。第2行首先输入B的元素个数n(1≤n≤100),然后输入n个元素。
输出格式:
对于每组测试数据。分别输出将A、B合并后的数组C的全部元素。输出的元素之间以一个空格分隔(最后一个数据之后没有空格)。
输入样例:
1
4 3 5 8 11
7 2 6 8 9 11 15 20
输出样例:
2 3 5 6 8 8 9 11 11 15 20
分析:
-
merge函数:这个函数接受两个已排序的数组a和b,返回一个新的已排序数组c,该数组是将a和b合并后排序的结果。
- 创建一个长度为a.length + b.length的新数组c。
- 使用三个索引i、j和k来遍历a、b和c。其中i和j分别用于追踪a和b的当前元素,而k用于追踪c的当前位置。
- 通过比较a[i]和b[j]的值,将较小的值添加到c[k],并将相应的索引递增。
- 如果a或b的元素已经被完全添加到c,但是另一个数组仍有元素未添加,那么将剩余的元素添加到c。
- 最后返回合并后的数组c。
-
main函数:这个函数从标准输入读取测试用例的数量t,然后对每个测试用例执行以下操作:
- 读取两个已排序的数组a和b。
- 调用merge函数将a和b合并。
- 打印合并后的数组。
Java语言:
import java.util.Scanner;
public class Main {
public static int[] merge(int[] a, int[] b) {
int[] c = new int[a.length + b.length];
int i = 0, j = 0, k = 0;
while (i < a.length && j < b.length) {
if (a[i] <= b[j]) {
c[k++] = a[i++];
} else {
c[k++] = b[j++];
}
}
while (i < a.length) {
c[k++] = a[i++];
}
while (j < b.length) {
c[k++] = b[j++];
}
return c;
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int t = sc.nextInt();
while (t-- > 0) {
int m = sc.nextInt();
int[] a = new int[m];
for (int i = 0; i < m; i++) {
a[i] = sc.nextInt();
}
int n = sc.nextInt();
int[] b = new int[n];
for (int i = 0; i < n; i++) {
b[i] = sc.nextInt();
}
int[] c = merge(a, b);
for (int i = 0; i < c.length; i++) {
System.out.print(c[i]);
if (i != c.length - 1) {
System.out.print(" ");
}
}
System.out.println();
}
}
}
总结:
整体而言,这是一个相对直接的程序,其核心是merge函数,该函数实现了两个已排序数组的合并操作。在主函数中,它主要用于处理多个测试用例,每个测试用例都需要输入两个已排序的数组,然后输出合并后的结果。