【华为OD机试2023】AI处理器组合Python

题目描述：

某公司研发了一款高性能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个的环。

解题思路：

逻辑题，用额外的空间存储亲和性调度原则，然后判断是否存在符合高优先级的链路，如果不存在的话，继续判断下一优先级的亲和性原则

qhx1=[1,3,2,4]
qhx2=[2,4,3]
lianlus={1:[],2:[]}
lst=eval(input())
for cpu in lst:
    if 0<=cpu<=3:
        lianlus[1].append(cpu)
    else:
        lianlus[2].append(cpu)
avalible=[]
cpu_1=len(lianlus[1])
cpu_2=len(lianlus[2])
need=int(input())
if cpu_2<need and cpu_1<need:
    print('[]')
else:
    if need==1:
        flag=False
        for qhx in qhx1:
            for id in lianlus.keys():
                lianlu=lianlus[id]
                if len(lianlu)==qhx:
                    flag=True
                    for cpu in lianlu:
                        avalible.append([cpu])
            if flag:
                break
        if avalible!=[]:
            print(avalible)
        else:
            print('[]')
        
    elif need==2:
        flag=False
        for qhx in qhx2:
            for id in lianlus.keys():
                lianlu=lianlus[id]
                if len(lianlu)==qhx:
                    flag=True
                    l=len(lianlu)
                    for i in range(l):
                        for j in range(i+1,l):
                            avalible.append([lianlu[i],lianlu[j]])
            if flag:
                break
        if avalible!=[]:
            print(avalible)
        else:
            print('[]')
    elif need==4:
        flag=False
        for id in lianlus.keys():
            lianlu=lianlus[id]
            if len(lianlu)==4:
                flag=True
                avalible.append(lianlu)
        if flag:
            print(avalible)
        else:
            print('[]')

    elif need==8:
            if len(lianlus[1])+len(lianlus[2]) == 8:
                flag = True
                print('[0,1,2,3,4,5,6,7,]')
            else:
                print('[]')