华为OD机试 - 最优策略组合下的总的系统消耗资源数（Java 2023 B卷 100分）

在这里插入图片描述

专栏导读

本专栏收录于《华为OD机试（JAVA）真题（A卷+B卷）》。

刷的越多，抽中的概率越大，每一题都有详细的答题思路、详细的代码注释、样例测试，发现新题目，随时更新，全天CSDN在线答疑。

一、题目描述

在通信系统中有一个常见的问题是对用户进行不同策略的调度，会得到不同系统消耗的性能。

假设由N个待串行用户，每个用户可以使用A/B/C三种不同的调度策略，不同的策略会消耗不同的系统资源。

请你根据如下规则进行用户调度，并返回总的消耗资源数。

规则是：相邻的用户不能使用相同的调度策略。

例如：

第一个用户使用A策略则第二个用户只能使用B和C策略；
对单的用户而言，不同的调度策略对系统资源的消耗可以规划后抽象为数值；

例如：

某用户分别使用ABC策略的系统消耗，分别为15 8 17；
每个用户依次选择当前所能选择的对系统资源消耗最少的策略,局部最优；
如果有多个满足要求的策略，选最后一个。

二、输入描述

第一行表示用户个数N。
接下来表示每一行表示一个用户分别使用三个策略的资源消耗，resA resB resC。

三、输出描述

最优策略组合下的总的系统消耗资源数。

四、解题思路

读取输入的用户数量 n；
创建一个列表 lists，用于存储每个用户的不同策略和资源消耗；
循环读取每个用户的资源消耗，将其添加到 lists 中；
初始化变量 total 表示总的系统消耗资源数，变量 preUser 表示上一个用户所选的策略；
对每个用户进行调度：
- 对当前用户的所有策略按照资源消耗从低到高排序，如果资源消耗相同，则将策略索引较大的排在前面，以满足选择最后一个的要求；
- 遍历排序后的策略，如果是第一个用户，或者当前策略与上一个用户所选的策略不同，选择该策略，并将其资源消耗加到 total 中，更新 preUser 为当前策略索引，然后跳出循环。
输出 total，即最优策略组合下的总的系统消耗资源数。

五、Java算法源码

public static void main(String[] args) {
    Scanner scanner = new Scanner(System.in);
    int n = scanner.nextInt();

    List<List<User>> lists = new ArrayList<List<User>>(n);
    for (int i = 0; i < n; i++) {
        List<User> iUserList = new ArrayList<>(3);
        for (int j = 0; j < 3; j++) {
            User User = new User();
            User.index = j;
            User.value = scanner.nextInt();
            iUserList.add(User);
        }
        lists.add(iUserList);
    }

    int total = 0;
    int preUser = -1;
    for (int i = 0; i < n; i++) {
        // 当前用户的所有策略，按照资源消耗从低到高排序，消耗一样多的，index大的排在前面
        List<User> iUserList = lists.get(i);
        iUserList = iUserList.stream()
                .sorted(Comparator.comparing(User::getIndex)) // 如果有多个满足要求的策略，选最后一个，所以排在前面供选择
                .sorted(Comparator.comparing(User::getValue)) // 后执行的优先级高
                .collect(Collectors.toList());
        for (User User : iUserList) {
            // 第一个用户，或者后面的用户和之前用户不是同一个
            if (preUser == -1 || User.getIndex() != preUser) {
                total += User.getValue();
                preUser = User.getIndex();
                break;
            }
        }
    }
    System.out.println(total);
}

private static class User {
    public int index;
    public int value;

    public int getIndex() {
        return index;
    }

    public void setIndex(int index) {
        this.index = index;
    }

    public int getValue() {
        return value;
    }

    public void setValue(int value) {
        this.value = value;
    }
}