华为机试：最优策略组合下的总的系统消耗资源数

题目描述

在通信系统中有一个常见的问题是对用户进行不同策略的调度
会得到不同系统消耗的性能
假设由N个待串行用户，每个用户可以使用A/B/C三种不同的调度策略
不同的策略会消耗不同的系统资源
请你根据如下规则进行用户调度
并返回总的消耗资源数
规则是：相邻的用户不能使用相同的调度策略
例如：
第一个用户使用A策略 则第二个用户只能使用B和C策略
对单的用户而言，不同的调度策略对系统资源的消耗可以规划后抽象为数值
例如
某用户分别使用ABC策略的系统消耗，分别为15 8 17
每个用户依次选择当前所能选择的对系统资源消耗最少的策略,局部最优
如果有多个满足要求的策略，选最后一个。

输入描述

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

输出描述

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

示例1

输入

3
15 8 17
12 20 9
11 7 5

输出

24

说明：

1号用户使用B策略
2号用户使用C策略
3号用户使用B策略
系统资源消耗8+9+7。

思路分析

• 每个用户可以使用A/B/C三种不同的调度策略，即固定3种。
• 每个用户依次选择当前所能选择的对系统资源消耗最少的策略，局部最优，每个用户的策略需要根据资源消耗进行排序。
• 相邻的用户不能使用相同的调度策略，即上一个用户选择的这个策略，当前用户就不能再选择。
• 如果有多个满足要求的策略，选最后一个。所以还要考虑不同策略消耗资源相等的情况。
• 遍历处理每个用户可选择的消耗最小的策略相加。
• 可以考虑面向对象编程，代码更易懂。

参考代码

注：题目网上找的，参考代码是练习用，仅供参考，并不保证用例通过率。

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

/**
 * 最优策略组合下的总的系统消耗资源数
 */
public class Test0021 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();

        List<List<Res>> resList = new ArrayList<>(n);
        for (int i = 0; i < n; i++) {
            List<Res> iResList = new ArrayList<>(3);
            for (int j = 0; j < 3; j++) {
                Res res = new Res();
                res.index = j;
                res.value = scanner.nextInt();
                iResList.add(res);
            }
            resList.add(iResList);
        }

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

    static class Res {
        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;
        }

        @Override
        public String toString() {
            return "Res{" +
                    "index=" + index +
                    ", value=" + value +
                    '}';
        }
    }
}