华为OD机试真题（Java），最优策略组合下的总的系统消耗资源数（100%通过+复盘思路）

news2024/11/17 8:54:01

在这里插入图片描述

一、题目描述

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

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

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

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

例如：

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

例如：

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

二、输入描述

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

三、输出描述

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

四、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;
    }
}