推荐多样性 - 华为OD统一考试(D卷)

OD统一考试（D卷）

分值： 200分

题解： Java / Python / C++

题目描述

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

1. 各个列表元素需要做穿插处理，即先从第一个列表中为每屏选择一个元素，再从第二个列表中为每屏选择一个元素，依次类推

2. 每个列表的元素尽量均分为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. 每个列表会保证元素数量满足窗口要求，不需要考虑元素不足情况

2. 每个列表的元素已去重，不需要考虑元素重复情况

3. 每个列表的元素列表均不为空，不需要考虑列表为空情况

4. 每个列表的元素列表已经过排序处理，输出结果要保证不改变同一个列表的元素顺序

5. 每个列表的元素数量可能是不同的

示例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

题解

这道题目是一个简单的数据处理问题，需要根据输入的窗口数量、窗口大小以及多个列表中的元素，按照规定的选择策略进行元素的选取和输出。下面是解题思路和代码描述：

解题思路：

• 首先读取输入的窗口数量和窗口大小，然后读取多个列表的元素。
• 根据窗口数量和窗口大小，依次从每个列表中选择元素并放入对应的窗口中，按照规定的选择策略进行穿插处理。
• 最后输出选取的元素列表，合并多个窗口的元素为一个列表输出。

Java

import java.util.*;

/**
 * @author code5bug
 */
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt(), k = in.nextInt();
        in.nextLine();  // 消耗掉换行符号

        // 读入列表元素
        List<Queue<Integer>> lq = new ArrayList<>();
        while (in.hasNext()) {
            int[] nums = Arrays.stream(in.nextLine().split(" "))
                    .mapToInt(Integer::parseInt).toArray();

            Queue<Integer> queue = new ArrayDeque<>();
            for (int num : nums) queue.add(num);
            lq.add(queue);
        }

        //