目录

题目描述

输入描述

输出描述

用例

题目解析

算法源码

题目描述

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

用例

题目解析

用例中，链路link1=[0,1]，链路link2=[4,5,6,7]

现在要选1个处理器，则需要按照亲和性调度原则

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

最佳的是，找剩余可用1个处理器的链路，发现没有，link1剩余可用2，link2剩余可用4

其次的是，找剩余可用3个处理器的链路，发现没有

再次的是，找剩余可用2个处理器的链路，link1符合要求，即从0和1处理器中任选一个，有两种选择，可以使用dfs找对应组合。

算法源码

/* JavaScript Node ACM模式 控制台输入获取 */
const readline = require("readline");const rl = readline.createInterface({input: process.stdin,output: process.stdout,
});const lines = [];
rl.on("line", (line) => {lines.push(line);if (lines.length === 2) {const arr = JSON.parse(lines[0]);const num = lines[1];console.log(getResult(arr, num));lines.length = 0;}
});function getResult(arr, num) {const link1 = [];const link2 = [];arr.sort((a, b) => a - b).forEach((e) => {e < 4 ? link1.push(e) : link2.push(e);});const ans = [];const len1 = link1.length;const len2 = link2.length;switch (num) {case "1":if (len1 === 1 || len2 === 1) {if (len1 === 1) dfs(link1, 0, 1, [], ans);if (len2 === 1) dfs(link2, 0, 1, [], ans);} else if (len1 === 3 || len2 === 3) {if (len1 === 3) dfs(link1, 0, 1, [], ans);if (len2 === 3) dfs(link2, 0, 1, [], ans);} else if (len1 === 2 || len2 === 2) {if (len1 === 2) dfs(link1, 0, 1, [], ans);if (len2 === 2) dfs(link2, 0, 1, [], ans);} else if (len1 === 4 || len2 === 4) {if (len1 === 4) dfs(link1, 0, 1, [], ans);if (len2 === 4) dfs(link2, 0, 1, [], ans);}break;case "2":if (len1 === 2 || len2 === 2) {if (len1 === 2) dfs(link1, 0, 2, [], ans);if (len2 === 2) dfs(link2, 0, 2, [], ans);} else if (len1 === 4 || len2 === 4) {if (len1 === 4) dfs(link1, 0, 2, [], ans);if (len2 === 4) dfs(link2, 0, 2, [], ans);} else if (len1 === 3 || len2 === 3) {if (len1 === 3) dfs(link1, 0, 2, [], ans);if (len2 === 3) dfs(link2, 0, 2, [], ans);}break;case "4":if (len1 === 4 || len2 === 4) {if (len1 === 4) ans.push(link1);if (len2 === 4) ans.push(link2);}break;case "8":if (len1 === 4 && len2 === 4) {ans.push([...link1, ...link2]);}break;}return JSON.stringify(ans);
}function dfs(arr, index, level, path, res) {if (path.length === level) {return res.push([...path]);}for (let i = index; i < arr.length; i++) {path.push(arr[i]);dfs(arr, i + 1, level, path, res);path.pop();}
}