[华为机试]某公司研发了一款高性能AI处理器

[华为机试]某公司研发了一款高性能AI处理器

题目描述

某公司研发了一款高性能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、2、4、8。 编号0-3的处理器处于一个链路，编号4-7的处理器处于另外一个链路。 处理器编号唯一，且不存在相同编号处理器。

做题思路

为了满足亲和性调度原则，我们可以分别考虑申请处理器个数为1、2、4、8时的情况。

对于申请处理器个数为1的情况，我们可以先遍历链路0-3，如果剩余可用的处理器数量为1个，则选择这个链路；如果不是1个，则遍历链路4-7，如果剩余可用的处理器数量为1个，则选择这个链路；如果剩余可用的处理器数量都不是1个，则返回空列表。

对于申请处理器个数为2的情况，我们可以先遍历链路0-3，如果剩余可用的处理器数量为2个或者4个，则选择这个链路；如果不是2个或者4个，则遍历链路4-7，如果剩余可用的处理器数量为2个或者4个，则选择这个链路；如果剩余可用的处理器数量都不是2个或者4个，则返回空列表。

对于申请处理器个数为4的情况，我们可以先遍历链路0-3，如果剩余可用的处理器数量为4个，则选择这个链路；如果不是4个，则遍历链路4-7，如果剩余可用的处理器数量为4个，则选择这个链路；如果剩余可用的处理器数量都不是4个，则返回空列表。

对于申请处理器个数为8的情况，我们可以直接返回所有处理器编号的列表。

function selectChips(array, num) {
  if (num === 1) {
    // 遍历链路0-3，如果剩余可用的处理器数量为1个，则选择这个链路
    for (let i = 0; i < 4; i++) {
      if (array.includes(i)) {
        return [i];
      }
    }
    // 遍历链路4-7，如果剩余可用的处理器数量为1个，则选择这个链路
    for (let i = 4; i < 8; i++) {
      if (array.includes(i)) {
        return [i];
      }
    }
    // 剩余可用的处理器数量都不是1个，则返回空列表
    return [];
  } else if (num === 2) {
    // 遍历链路0-3，如果剩余可用的处理器数量为2个或者4个，则选择这个链路
    for (let i = 0; i < 4; i++) {
      if (array.includes(i)) {
        for (let j = i + 1; j < 4; j++) {
          if (array.includes(j)) {
            return [i, j];
          }
        }
      }
    }
    // 遍历链路4-7，如果剩余可用的处理器数量为2个或者4个，则选择这个链路
    for (let i = 4; i < 8; i++) {
      if (array.includes(i)) {
        for (let j = i + 1; j < 8; j++) {
          if (array.includes(j)) {
            return [i, j];
          }
        }
      }
    }
    // 剩余可用的处理器数量都不是2个或者4个，则返回空列表
    return [];
  } else if (num === 4) {
    // 遍历链路0-3，如果剩余可用的处理器数量为4个，则选择这个链路
    for (let i = 0; i < 4; i++) {
      if (array.includes(i)) {
        for (let j = i + 1; j < 4; j++) {
          if (array.includes(j)) {
            for (let k = j + 1; k < 4; k++) {
              if (array.includes(k)) {
                for (let l = k + 1; l < 4; l++) {
                  if (array.includes(l)) {
                    return [i, j, k, l];
                  }
                }
              }
            }
          }
        }
      }
    }
    // 遍历链路4-7，如果剩余可用的处理器数量为4个，则选择这个链路
    for (let i = 4; i < 8; i++) {
      if (array.includes(i)) {
        for (let j = i + 1; j < 8; j++) {
          if (array.includes(j)) {
            for (let k = j + 1; k < 8; k++) {
              if (array.includes(k)) {
                for (let l = k + 1; l < 8; l++) {
                  if (array.includes(l)) {
                    return [i, j, k, l];
                  }
                }
              }
            }
          }
        }
      }
    }
    // 剩余可用的处理器数量都不是4个，则返回空列表
    return [];
  } else if (num === 8) {
    return array;
  }
}


// 示例
const array = [0, 1, 2, 3, 4, 5, 6, 7];
let num = 1;
console.log(selectChips(array, num)); // 输出：[0]
num = 2;
console.log(selectChips(array, num)); // 输出：[0, 1]
num = 4;
console.log(selectChips(array, num)); // 输出：[0, 1, 2, 3]
num = 8;
console.log(selectChips(array, num)); // 输出：[0, 1, 2, 3, 4, 5, 6, 7]