OD机试真题-数组拼接

题目描述

现在有多组整数数组，需要将它们合并成一个新的数组。

合并规则，从每个数组里按顺序取出固定长度的内容合并到新的数组中，取完的内容会删除掉，如果该行不足固定长度或者已经为空，则直接取出剩余部分的内容放到新的数组中，继续下一行。

输入描述

第一行是每次读取的固定长度，0<长度<10

第二行是整数数组的数目，0<数目<1000

第3-n行是需要合并的数组，，不同的数组用回车换行分隔，数组内部用逗号分隔，最大不超过100个元素。

输出描述

输出一个新的数组，用逗号分隔。

示例1

输入

3
2
2,5,6,7,9,5,7
1,7.4 3,4

输出

2,5,6,1,7,4,7,9,5,3,4,7

说明

获得长度3和数组数目2
先遍历第一行，获得2,5,6
再遍历第二行，获得1,7.4
再循环回到第一行，获得7,9,5
再遍历第二行，获得3.4
再回到第一行，获得7，按顺序拼接成最终结果

示例2

输入

4
3
1,2,3,4,5,6
1,2,3
1,2,3,4

输出

1,2 ,3 ,4,1, 2, 3,1,2,3,4,5,6

题解

使用 List<List<Integer>> 对数据进行存储

1. 通过输入得出最大数组的长度
2. 遍历List
3. 第一次取 0~固定长度的数据 第二次取 固定长度~2倍固定长度的数据一次类推 直到大于最大数组的长度

源码 Java

public class ArrayConcat {

	static Input input;
	static {
		input = new Input("4\n" +
				"3\n" +
				"1,2,3,4,5,6\n" +
				"1,2,3\n" +
				"1,2,3,4");
	}

	public static void main(String[] args) {
		int subLen = Integer.parseInt(input.nextLine());
		int n = Integer.parseInt(input.nextLine());
		List<List<String>> list = new ArrayList<>();
		List<String> results = new ArrayList<>();

		int maxLen = 0;
		for (int i = 0; i < n; i++) {
			List<String> lines = new ArrayList<>();
			String[] split = input.nextLine().split(",");
			for (int j = 0; j < split.length; j++) {
				lines.add(split[j]);
			}
			maxLen = Math.max(maxLen, lines.size());
			list.add(lines);
		}
		int subIndex = 0;
		while (subIndex < maxLen) {
			for (int i = 0; i < list.size(); i++) {
				List<String> line = list.get(i);
				int start = subIndex;
				int end = subIndex + subLen;
				while (start < end && start < line.size()) {
					results.add(line.get(start++));
				}
			}
			subIndex += subLen;
		}

		System.out.println(String.join(",", results));
	}
}