2023华为OD机试真题-AI处理器组合(JAVA、Python、C++)

本文链接：https://blog.youkuaiyun.com/huaweiod123/article/details/130610705

该博客介绍了华为在线开发者测试中的一道真题，涉及AI处理器的亲和性调度策略。问题要求根据给定的处理器编号数组和任务需求，找出满足特定条件的处理器组合。解决方案需考虑不同申请数量下的最佳链路选择，如单个处理器优先选择剩余1个的链路等。示例展示了不同输入和输出的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

某公司研发了一款高性能AI处理器。每台物理设备具备8颗AI处理器，编号分别为0、1、2、3、4、5、6、7。编号0-3的处理器处于同一个链路中，编号4-7的处理器处于另外一个链路中，不通链路中的处理器不能通信，如下图所示。现给定服务器可用的处理器编号数组array，以及任务申请的处理器数量num，找出符合下列亲和性调度原则的芯片组合。如果不存在符合要求的组合，则返回空列表。

亲和性调度原则：

- 如果申请处理器个数为1，则选择同一链路，剩余可用的处理器数量为1个的最佳，其次是剩余3个的为次佳，然后是剩余2个，最后是剩余4个。

- 如果申请处理器个数为2，则选择同一链路剩余可用的处理器数量2个的为最佳，其次是剩余4个，最后是剩余3个。

- 如果申请处理器个数为4，则必须选择同一链路剩余可用的处理器数量为4个。

- 如果申请处理器个数为8，则申请节点所有8个处理器。

提示：

1. 任务申请的处理器数量只能是1、2、4、8

2. 编号0-3的处理器处于一个链路，编号4-7的处理器处于另外一个链路。

3. 处理器编号唯一，且不存在相同编号处理器

入描述：

输入包含可用的处理器编号数组array，以及任务申请的处理器数量num两个部分。

第一行为array，第二行为num。例如：

[0, 1, 4, 5, 6, 7]

1

表示当前编号为0、1、4、5、6、7的处理器可用。任务申请1个处理器。

0<= array.length <= 8

0<= array[i] <= 7

num in [1, 2, 4, 8]

输出描述：

输出为组合列表，当array=[0, 1, 4, 5, 6, 7] ，num=1时，输出为[[0], [1]]

示例1

输入：

[0, 1, 4, 5, 6, 7]
1
输出：

[[0], [1]]
说明：

根据第一条亲和性调度原则，在剩余两个处理器的链路（0,1,2,3）中选择处理器。由于只有0和1可用，则返回任意一颗处理器即可

示例2

输入：

[0, 1, 4, 5, 6, 7]
4
输出：

[[4, 5, 6, 7]]
说明：

根据第三条亲和性调度原则，必须选择同一链路剩余可用的处理器数量为4个的环。

#include<iostream>
#include<vector>
#include<string>
using namespace std;
int main(){
    vector<int> arr1,arr2;
    int num;
    string str;
    getline(cin,str);
    for(int i=0;i<str.length();++i){
        if (str.at(i) >= '0' && str.at(i) <= '3'){
            arr1.push_back(str.at(i) - '0');
        } else if (str.at(i) >= '4' && str.at(i) <= '7'){
            arr2.push_back(str.at(i) - '0');
        } else continue;
    }
    cin>>num;
 
    switch(num){
        case 1:
            {
                bool flag =false;
                if (arr1.size() == 1) {
                    cout<<"[["<<arr1[0]<<"]";
                    flag = true;
                }
                if (arr2.size() == 1) {
                    if (flag) cout<<", ["<<arr2[0]<<"]]";
                    else {cout<<"[["<<arr2[0]<<"]";flag = true;}
                }
                if(flag) {cout<<"]";break;}
 
                if (arr1.size() == 3) {
                    cout<<"[["<<arr1[0]<<"]&#