华为OD机试真题-推荐多样性

题目描述

推荐多样性需要从多个列表中选择元素，一次性要返回N屏数据（窗口数量），每屏展示K个元素（窗口大小），选择策略：

各个列表元素需要做穿插处理，即先从第一个列表中为每屏选择一个元素，再从第二个列表中为每屏选择一个元素，依次类推
每个列表的元素尽量均分为N份，如果不够N个，也要全部分配完，参考样例图：

（1）从第一个列表中选择4条0 1 2 3，分别放到4个窗口中

（2）从第二个列表中选择4条10 11 12 13，分别放到4个窗口中

（3）从第三个列表中选择4条20 21 22 23，分别放到4个窗口中

（4）再从第一个列表中选择4条4 5 6 7，分别放到4个窗口中

（5）再从第一个列表中选择，由于数量不足4条，取剩下的2条，放到窗口1和窗口2

（6）再从第二个列表中选择，由于数量不足4条并且总的元素数达到窗口要求，取18 19放到窗口3和窗口4

输入描述

第一行输入为N，表示需要输出的窗口数量，取值范围 [1, 10]

第二行输入为K，表示每个窗口需要的元素数量，取值范围 [1, 100]

之后的行数不定（行数取值范围[1,10]），表示每个列表输出的元素列表。元素之间以空格分隔，已经过排序处理，每个列表输出的元素数量取值范围[1,100]

输出描述：

输出元素列表，元素数量=窗口数量*窗口大小，元素之间以空格分隔，多个窗口合并为一个列表输出，参考样例：

先输出窗口1的元素列表，再输出窗口2的元素列表，再输出窗口3的元素列表，最后输出窗口4的元素列表

补充说明：

每个列表会保证元素数量满足窗口要求，不需要考虑元素不足情况
每个列表的元素已去重，不需要考虑元素重复情况
每个列表的元素列表均不为空，不需要考虑列表为空情况
每个列表的元素列表已经过排序处理，输出结果要保证不改变同一个列表的元素顺序
每个列表的元素数量可能是不同的

示例1

输入：

4
7
0 1 2 3 4 5 6 7 8 9
10 11 12 13 14 15 16 17 18 19
20 21 22 23 24 25 26 27 28 29
输出：

0 10 20 4 14 24 8 1 11 21 5 15 25 9 2 12 22 6 16 26 18 3 13 23 7 17 27 19
说明

题解

感觉这个题目有歧义
假如有一行被取走了三个元素，还有4个元素，下一次在该行取四个元素，还是取一个元素，没有说明

源码 Java

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;public class Windows {static Input input;static {input = new Input("4\n" +"7\n" +"0 1 2 3 4 5 6 7 8 9\n" +"10 11 12 13 14 15 16 17 18 19\n" +"20 21 22 23 24 25 26 27 28 29");}public static void main(String[] args) {int windowSize = Integer.parseInt(input.nextLine());int content = Integer.parseInt(input.nextLine());String[][] matrix = new String[windowSize][content];List<List<String>> list = new ArrayList<>();while (input.hasNextLine()) {String line = input.nextLine();String[] split = line.split(" ");List<String> list1 = Arrays.asList(split);list.add(list1);}int[] index = new int[list.size()];int count = 0;for (int i = 0; i < content; i++) {int old = count;for (int j = 0; j < windowSize; j++) {while (index[count] >= list.get(count).size()) {count++;count = count % list.size();}matrix[j][i] = list.get(count).get(index[count]);index[count]++;}if (count == old) {// 移动到下一行count++;count = count % list.size();while (index[count] >= list.get(count).size()) {count++;count = count % list.size();}}}for (int i = 0; i < windowSize; i++) {for (int j = 0; j < content; j++) {System.out.print(matrix[i][j] + " ");}}}}