【2024年华为OD机试】(A卷,100分)- 处理器问题(Java JS PythonC/C++)

ops/2025/1/12 13:10:51/

在这里插入图片描述

一、问题描述

题目描述

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

输入描述

输入包含可用的处理器编号数组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]]。

用例

输入

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

输出

[[0], [1]]

说明

根据第一条亲和性调度原则,在剩余两个处理器的链路(0, 1, 2, 3)中选择处理器。

由于只有0和1可用,则返回任意一颗处理器即可。

输入

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

输出

[[4, 5, 6, 7]]

说明

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

题目解析

用例中,链路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算法源码

以下是 JavaScript 代码的中文详细注释和逻辑讲解:


代码逻辑

javascript">/* JavaScript Node ACM模式 控制台输入获取 */
const readline = require("readline");// 创建 readline 接口,用于从控制台读取输入
const rl = readline.createInterface({input: process.stdin,output: process.stdout,
});// 存储输入的行数据
const lines = [];// 监听输入事件
rl.on("line", (line) => {lines.push(line); // 将输入的行数据存入 lines 数组// 当输入的行数达到 2 行时,开始处理数据if (lines.length === 2) {const arr = JSON.parse(lines[0]); // 解析第一行输入为数组const num = lines[1]; // 获取第二行输入的数字// 调用算法函数并输出结果console.log(getResult(arr, num));// 清空 lines 数组,准备接收下一组输入lines.length = 0;}
});// 算法入口
function getResult(arr, num) {const link1 = []; // 存储小于 4 的元素const link2 = []; // 存储大于等于 4 的元素// 将数组排序并分为两组arr.sort((a, b) => a - b) // 升序排序.forEach((e) => {e < 4 ? link1.push(e) : link2.push(e); // 根据元素大小分组});const ans = []; // 存储最终结果const len1 = link1.length; // link1 的长度const len2 = link2.length; // link2 的长度// 根据输入的数字 num 执行不同的逻辑switch (num) {case "1":// 如果 link1 或 link2 的长度为 1,调用 dfs 函数if (len1 === 1 || len2 === 1) {if (len1 === 1) dfs(link1, 0, 1, [], ans);if (len2 === 1) dfs(link2, 0, 1, [], ans);}// 如果 link1 或 link2 的长度为 3,调用 dfs 函数else if (len1 === 3 || len2 === 3) {if (len1 === 3) dfs(link1, 0, 1, [], ans);if (len2 === 3) dfs(link2, 0, 1, [], ans);}// 如果 link1 或 link2 的长度为 2,调用 dfs 函数else if (len1 === 2 || len2 === 2) {if (len1 === 2) dfs(link1, 0, 1, [], ans);if (len2 === 2) dfs(link2, 0, 1, [], ans);}// 如果 link1 或 link2 的长度为 4,调用 dfs 函数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":// 如果 link1 或 link2 的长度为 2,调用 dfs 函数if (len1 === 2 || len2 === 2) {if (len1 === 2) dfs(link1, 0, 2, [], ans);if (len2 === 2) dfs(link2, 0, 2, [], ans);}// 如果 link1 或 link2 的长度为 4,调用 dfs 函数else if (len1 === 4 || len2 === 4) {if (len1 === 4) dfs(link1, 0, 2, [], ans);if (len2 === 4) dfs(link2, 0, 2, [], ans);}// 如果 link1 或 link2 的长度为 3,调用 dfs 函数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":// 如果 link1 或 link2 的长度为 4,直接将整个数组加入结果if (len1 === 4 || len2 === 4) {if (len1 === 4) ans.push(link1);if (len2 === 4) ans.push(link2);}break;case "8":// 如果 link1 和 link2 的长度都为 4,将两个数组合并后加入结果if (len1 === 4 && len2 === 4) {ans.push([...link1, ...link2]);}break;}// 将结果数组转换为字符串,并用逗号分隔return JSON.stringify(ans).split(",").join(", ");
}// 深度优先搜索(DFS)函数
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(); // 回溯,移除当前元素}
}

代码讲解

  1. 输入处理

    • 使用 readline 模块从控制台读取输入。
    • 将输入的行数据存入 lines 数组。
    • 当输入的行数达到 2 行时,解析第一行为数组 arr,第二行为字符串 num
  2. 数据分组

    • 将数组 arr 排序并分为两组:
      • link1:存储小于 4 的元素。
      • link2:存储大于等于 4 的元素。
  3. 逻辑分支

    • 根据输入的数字 num,执行不同的逻辑:
      • num === "1":处理长度为 1、2、3、4 的情况,调用 dfs 函数生成组合。
      • num === "2":处理长度为 2、3、4 的情况,调用 dfs 函数生成组合。
      • num === "4":处理长度为 4 的情况,直接将整个数组加入结果。
      • num === "8":处理 link1link2 长度都为 4 的情况,将两个数组合并后加入结果。
  4. 深度优先搜索(DFS)

    • dfs 函数用于生成指定长度的组合。
    • 通过递归和回溯,生成所有可能的组合,并将结果存入 res 数组。
  5. 结果输出

    • 将结果数组 ans 转换为字符串,并用逗号分隔后返回。

示例解析

输入
[1, 2, 3, 4, 5, 6, 7, 8]
"2"
运行结果
[[1, 2], [1, 3], [2, 3], [4, 5], [4, 6], [5, 6]]
  • 解析:
    • 输入数组 [1, 2, 3, 4, 5, 6, 7, 8] 被分为 link1 = [1, 2, 3]link2 = [4, 5, 6, 7, 8]
    • 输入 num = "2",表示需要生成长度为 2 的组合。
    • 最终结果为 link1link2 中长度为 2 的所有组合。

总结

  • 该代码通过分组和深度优先搜索(DFS),实现了根据输入条件生成指定长度的组合。
  • 适用于需要处理分组和组合问题的场景,尤其是需要根据条件动态生成结果的场景。

如果有其他问题,欢迎随时提问!

三、Java算法源码

以下是 Java 代码的中文详细注释和逻辑讲解:


代码逻辑

java">import java.util.ArrayList;
import java.util.Arrays;
import java.util.Scanner;
import java.util.stream.Collectors;
import java.util.stream.Stream;public class Main {public static void main(String[] args) {// 创建 Scanner 对象,用于从控制台读取输入Scanner sc = new Scanner(System.in);// 读取输入的第一行,解析为整数数组Integer[] arr =Arrays.stream(sc.nextLine().split("[\\[\\]\\,\\s]")) // 使用正则表达式分割输入.filter(str -> !"".equals(str)) // 过滤空字符串.map(Integer::parseInt) // 将字符串转换为整数.toArray(Integer[]::new); // 转换为 Integer 数组// 读取输入的第二行,作为条件参数String num = sc.next();// 调用算法函数并输出结果System.out.println(getResult(arr, num));}// 算法入口public static String getResult(Integer[] arr, String num) {// 定义两个列表,分别存储小于 4 和大于等于 4 的元素ArrayList<Integer> link1 = new ArrayList<>();ArrayList<Integer> link2 = new ArrayList<>();// 对数组进行排序Arrays.sort(arr, (a, b) -> a - b);// 遍历数组,将元素分为两组for (Integer e : arr) {if (e < 4) {link1.add(e); // 小于 4 的元素加入 link1} else {link2.add(e); // 大于等于 4 的元素加入 link2}}// 存储最终结果的列表ArrayList<ArrayList<Integer>> ans = new ArrayList<>();// 获取 link1 和 link2 的长度int len1 = link1.size();int len2 = link2.size();// 根据输入的条件 num 执行不同的逻辑switch (num) {case "1":// 如果 link1 或 link2 的长度为 1,调用 dfs 函数if (len1 == 1 || len2 == 1) {if (len1 == 1) dfs(link1, 0, 1, new ArrayList<>(), ans);if (len2 == 1) dfs(link2, 0, 1, new ArrayList<>(), ans);}// 如果 link1 或 link2 的长度为 3,调用 dfs 函数else if (len1 == 3 || len2 == 3) {if (len1 == 3) dfs(link1, 0, 1, new ArrayList<>(), ans);if (len2 == 3) dfs(link2, 0, 1, new ArrayList<>(), ans);}// 如果 link1 或 link2 的长度为 2,调用 dfs 函数else if (len1 == 2 || len2 == 2) {if (len1 == 2) dfs(link1, 0, 1, new ArrayList<>(), ans);if (len2 == 2) dfs(link2, 0, 1, new ArrayList<>(), ans);}// 如果 link1 或 link2 的长度为 4,调用 dfs 函数else if (len1 == 4 || len2 == 4) {if (len1 == 4) dfs(link1, 0, 1, new ArrayList<>(), ans);if (len2 == 4) dfs(link2, 0, 1, new ArrayList<>(), ans);}break;case "2":// 如果 link1 或 link2 的长度为 2,调用 dfs 函数if (len1 == 2 || len2 == 2) {if (len1 == 2) dfs(link1, 0, 2, new ArrayList<>(), ans);if (len2 == 2) dfs(link2, 0, 2, new ArrayList<>(), ans);}// 如果 link1 或 link2 的长度为 4,调用 dfs 函数else if (len1 == 4 || len2 == 4) {if (len1 == 4) dfs(link1, 0, 2, new ArrayList<>(), ans);if (len2 == 4) dfs(link2, 0, 2, new ArrayList<>(), ans);}// 如果 link1 或 link2 的长度为 3,调用 dfs 函数else if (len1 == 3 || len2 == 3) {if (len1 == 3) dfs(link1, 0, 2, new ArrayList<>(), ans);if (len2 == 3) dfs(link2, 0, 2, new ArrayList<>(), ans);}break;case "4":// 如果 link1 或 link2 的长度为 4,直接将整个列表加入结果if (len1 == 4 || len2 == 4) {if (len1 == 4) ans.add(link1);if (len2 == 4) ans.add(link2);}break;case "8":// 如果 link1 和 link2 的长度都为 4,将两个列表合并后加入结果if (len1 == 4 && len2 == 4) {ans.add(Stream.concat(link1.stream(), link2.stream()).collect(Collectors.toCollection(ArrayList<Integer>::new)));}break;}// 将结果列表转换为字符串并返回return ans.toString();}// 深度优先搜索(DFS)函数public static void dfs(ArrayList<Integer> arr, // 当前处理的列表int index, // 当前处理的起始索引int level, // 目标组合的长度ArrayList<Integer> path, // 当前路径ArrayList<ArrayList<Integer>> res // 结果列表) {// 如果当前路径的长度等于目标长度,将路径加入结果if (path.size() == level) {res.add(new ArrayList<>(path));return;}// 遍历数组,递归生成组合for (int i = index; i < arr.size(); i++) {path.add(arr.get(i)); // 将当前元素加入路径dfs(arr, i + 1, level, path, res); // 递归调用path.remove(path.size() - 1); // 回溯,移除当前元素}}
}

代码讲解

  1. 输入处理

    • 使用 Scanner 从控制台读取输入。
    • 第一行输入解析为整数数组 arr,使用正则表达式 [\\[\\]\\,\\s] 分割输入字符串,并过滤空字符串。
    • 第二行输入作为条件参数 num
  2. 数据分组

    • 将数组 arr 排序并分为两组:
      • link1:存储小于 4 的元素。
      • link2:存储大于等于 4 的元素。
  3. 逻辑分支

    • 根据输入的条件 num,执行不同的逻辑:
      • num == "1":处理长度为 1、2、3、4 的情况,调用 dfs 函数生成组合。
      • num == "2":处理长度为 2、3、4 的情况,调用 dfs 函数生成组合。
      • num == "4":处理长度为 4 的情况,直接将整个列表加入结果。
      • num == "8":处理 link1link2 长度都为 4 的情况,将两个列表合并后加入结果。
  4. 深度优先搜索(DFS)

    • dfs 函数用于生成指定长度的组合。
    • 通过递归和回溯,生成所有可能的组合,并将结果存入 res 列表。
  5. 结果输出

    • 将结果列表 ans 转换为字符串并返回。

示例解析

输入
[1, 2, 3, 4, 5, 6, 7, 8]
"2"
运行结果
[[1, 2], [1, 3], [2, 3], [4, 5], [4, 6], [5, 6]]
  • 解析:
    • 输入数组 [1, 2, 3, 4, 5, 6, 7, 8] 被分为 link1 = [1, 2, 3]link2 = [4, 5, 6, 7, 8]
    • 输入 num = "2",表示需要生成长度为 2 的组合。
    • 最终结果为 link1link2 中长度为 2 的所有组合。

总结

  • 该代码通过分组和深度优先搜索(DFS),实现了根据输入条件生成指定长度的组合。
  • 适用于需要处理分组和组合问题的场景,尤其是需要根据条件动态生成结果的场景。

如果有其他问题,欢迎随时提问!

四、Python算法源码

以下是 Python 代码的中文详细注释和逻辑讲解:


代码逻辑

python"># 输入获取
arr = eval(input())  # 读取输入的第一行,解析为列表
num = int(input())   # 读取输入的第二行,作为条件参数# 算法入口
def getResult(arr, num):# 定义两个列表,分别存储小于 4 和大于等于 4 的元素link1 = []link2 = []# 对数组进行排序arr.sort()# 遍历数组,将元素分为两组for e in arr:if e < 4:link1.append(e)  # 小于 4 的元素加入 link1else:link2.append(e)  # 大于等于 4 的元素加入 link2# 存储最终结果的列表ans = []# 获取 link1 和 link2 的长度len1 = len(link1)len2 = len(link2)# 根据输入的条件 num 执行不同的逻辑if num == 1:# 如果 link1 或 link2 的长度为 1,调用 dfs 函数if len1 == 1 or len2 == 1:if len1 == 1:dfs(link1, 0, 1, [], ans)if len2 == 1:dfs(link2, 0, 1, [], ans)# 如果 link1 或 link2 的长度为 3,调用 dfs 函数elif len1 == 3 or len2 == 3:if len1 == 3:dfs(link1, 0, 1, [], ans)if len2 == 3:dfs(link2, 0, 1, [], ans)# 如果 link1 或 link2 的长度为 2,调用 dfs 函数elif len1 == 2 or len2 == 2:if len1 == 2:dfs(link1, 0, 1, [], ans)if len2 == 2:dfs(link2, 0, 1, [], ans)# 如果 link1 或 link2 的长度为 4,调用 dfs 函数elif len1 == 4 or len2 == 4:if len1 == 4:dfs(link1, 0, 1, [], ans)if len2 == 4:dfs(link2, 0, 1, [], ans)elif num == 2:# 如果 link1 或 link2 的长度为 2,调用 dfs 函数if len1 == 2 or len2 == 2:if len1 == 2:dfs(link1, 0, 2, [], ans)if len2 == 2:dfs(link2, 0, 2, [], ans)# 如果 link1 或 link2 的长度为 4,调用 dfs 函数elif len1 == 4 or len2 == 4:if len1 == 4:dfs(link1, 0, 2, [], ans)if len2 == 4:dfs(link2, 0, 2, [], ans)# 如果 link1 或 link2 的长度为 3,调用 dfs 函数elif len1 == 3 or len2 == 3:if len1 == 3:dfs(link1, 0, 2, [], ans)if len2 == 3:dfs(link2, 0, 2, [], ans)elif num == 4:# 如果 link1 或 link2 的长度为 4,直接将整个列表加入结果if len1 == 4 or len2 == 4:if len1 == 4:ans.append(link1)if len2 == 4:ans.append(link2)elif num == 8:# 如果 link1 和 link2 的长度都为 4,将两个列表合并后加入结果if len1 == 4 and len2 == 4:tmp = []tmp.extend(link1)tmp.extend(link2)ans.append(tmp)# 返回结果列表return ans# 深度优先搜索(DFS)函数
def dfs(arr, index, level, path, res):# 如果当前路径的长度等于目标长度,将路径加入结果if len(path) == level:res.append(path[:])  # 使用 path[:] 创建 path 的副本return# 遍历数组,递归生成组合for i in range(index, len(arr)):path.append(arr[i])  # 将当前元素加入路径dfs(arr, i + 1, level, path, res)  # 递归调用path.pop()  # 回溯,移除当前元素# 算法调用
print(getResult(arr, num))

代码讲解

  1. 输入处理

    • 使用 input() 函数读取输入。
    • 第一行输入通过 eval() 解析为列表 arr
    • 第二行输入作为条件参数 num
  2. 数据分组

    • 将数组 arr 排序并分为两组:
      • link1:存储小于 4 的元素。
      • link2:存储大于等于 4 的元素。
  3. 逻辑分支

    • 根据输入的条件 num,执行不同的逻辑:
      • num == 1:处理长度为 1、2、3、4 的情况,调用 dfs 函数生成组合。
      • num == 2:处理长度为 2、3、4 的情况,调用 dfs 函数生成组合。
      • num == 4:处理长度为 4 的情况,直接将整个列表加入结果。
      • num == 8:处理 link1link2 长度都为 4 的情况,将两个列表合并后加入结果。
  4. 深度优先搜索(DFS)

    • dfs 函数用于生成指定长度的组合。
    • 通过递归和回溯,生成所有可能的组合,并将结果存入 res 列表。
  5. 结果输出

    • 将结果列表 ans 返回并打印。

示例解析

输入
[1, 2, 3, 4, 5, 6, 7, 8]
2
运行结果
[[1, 2], [1, 3], [2, 3], [4, 5], [4, 6], [5, 6]]
  • 解析:
    • 输入数组 [1, 2, 3, 4, 5, 6, 7, 8] 被分为 link1 = [1, 2, 3]link2 = [4, 5, 6, 7, 8]
    • 输入 num = 2,表示需要生成长度为 2 的组合。
    • 最终结果为 link1link2 中长度为 2 的所有组合。

总结

  • 该代码通过分组和深度优先搜索(DFS),实现了根据输入条件生成指定长度的组合。
  • 适用于需要处理分组和组合问题的场景,尤其是需要根据条件动态生成结果的场景。

如果有其他问题,欢迎随时提问!

五、C/C++算法源码:

以下是 C++ 代码的中文详细注释和逻辑讲解:


代码逻辑

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#include <sstream>
#include <iterator>
using namespace std;// 输入获取
vector<int> arr; // 存储输入数组
int num;         // 存储输入的条件参数// 算法入口
vector<vector<int>> getResult(const vector<int>& arr, int num) {// 定义两个列表,分别存储小于 4 和大于等于 4 的元素vector<int> link1, link2;// 遍历数组,将元素分为两组for (int e : arr) {if (e < 4) {link1.push_back(e); // 小于 4 的元素加入 link1} else {link2.push_back(e); // 大于等于 4 的元素加入 link2}}// 存储最终结果的列表vector<vector<int>> ans;// 获取 link1 和 link2 的长度int len1 = link1.size();int len2 = link2.size();// 根据输入的条件 num 执行不同的逻辑if (num == 1) {// 如果 link1 或 link2 的长度为 1,调用 dfs 函数if (len1 == 1) dfs(link1, 0, 1, {}, ans);if (len2 == 1) dfs(link2, 0, 1, {}, ans);// 如果 link1 或 link2 的长度为 3,调用 dfs 函数if (len1 == 3) dfs(link1, 0, 1, {}, ans);if (len2 == 3) dfs(link2, 0, 1, {}, ans);// 如果 link1 或 link2 的长度为 2,调用 dfs 函数if (len1 == 2) dfs(link1, 0, 1, {}, ans);if (len2 == 2) dfs(link2, 0, 1, {}, ans);// 如果 link1 或 link2 的长度为 4,调用 dfs 函数if (len1 == 4) dfs(link1, 0, 1, {}, ans);if (len2 == 4) dfs(link2, 0, 1, {}, ans);} else if (num == 2) {// 如果 link1 或 link2 的长度为 2,调用 dfs 函数if (len1 == 2) dfs(link1, 0, 2, {}, ans);if (len2 == 2) dfs(link2, 0, 2, {}, ans);// 如果 link1 或 link2 的长度为 4,调用 dfs 函数if (len1 == 4) dfs(link1, 0, 2, {}, ans);if (len2 == 4) dfs(link2, 0, 2, {}, ans);// 如果 link1 或 link2 的长度为 3,调用 dfs 函数if (len1 == 3) dfs(link1, 0, 2, {}, ans);if (len2 == 3) dfs(link2, 0, 2, {}, ans);} else if (num == 4) {// 如果 link1 或 link2 的长度为 4,直接将整个列表加入结果if (len1 == 4) ans.push_back(link1);if (len2 == 4) ans.push_back(link2);} else if (num == 8) {// 如果 link1 和 link2 的长度都为 4,将两个列表合并后加入结果if (len1 == 4 && len2 == 4) {vector<int> tmp;tmp.insert(tmp.end(), link1.begin(), link1.end());tmp.insert(tmp.end(), link2.begin(), link2.end());ans.push_back(tmp);}}// 去重处理set<vector<int>> uniqueAns;for (const auto& vec : ans) {uniqueAns.insert(vec);}ans.assign(uniqueAns.begin(), uniqueAns.end());// 返回结果列表return ans;
}// 深度优先搜索(DFS)函数
void dfs(const vector<int>& arr, int index, int level, vector<int> path, vector<vector<int>>& res) {// 如果当前路径的长度等于目标长度,将路径加入结果if (path.size() == level) {res.push_back(path);return;}// 遍历数组,递归生成组合for (int i = index; i < arr.size(); ++i) {path.push_back(arr[i]); // 将当前元素加入路径dfs(arr, i + 1, level, path, res); // 递归调用path.pop_back(); // 回溯,移除当前元素}
}// 算法调用
int main() {// 读取输入的第一行,解析为整数数组string input;getline(cin, input);istringstream iss(input);arr = vector<int>(istream_iterator<int>(iss), istream_iterator<int>());// 读取输入的第二行,作为条件参数cin >> num;// 调用算法函数并获取结果vector<vector<int>> result = getResult(arr, num);// 输出结果for (size_t i = 0; i < result.size(); ++i) {cout << '[';for (size_t j = 0; j < result[i].size(); ++j) {cout << result[i][j];if (j < result[i].size() - 1) cout << ", ";}cout << ']';if (i < result.size() - 1) cout << ", ";}cout << endl;return 0;
}

代码讲解

  1. 输入处理

    • 使用 getline 读取输入的第一行,并通过 istringstream 解析为整数数组 arr
    • 使用 cin 读取输入的第二行,作为条件参数 num
  2. 数据分组

    • 将数组 arr 分为两组:
      • link1:存储小于 4 的元素。
      • link2:存储大于等于 4 的元素。
  3. 逻辑分支

    • 根据输入的条件 num,执行不同的逻辑:
      • num == 1:处理长度为 1、2、3、4 的情况,调用 dfs 函数生成组合。
      • num == 2:处理长度为 2、3、4 的情况,调用 dfs 函数生成组合。
      • num == 4:处理长度为 4 的情况,直接将整个列表加入结果。
      • num == 8:处理 link1link2 长度都为 4 的情况,将两个列表合并后加入结果。
  4. 深度优先搜索(DFS)

    • dfs 函数用于生成指定长度的组合。
    • 通过递归和回溯,生成所有可能的组合,并将结果存入 res 列表。
  5. 去重处理

    • 使用 set<vector<int>> 对结果进行去重,确保结果中不包含重复的组合。
  6. 结果输出

    • 将结果列表 result 格式化输出。

示例解析

输入
1 2 3 4 5 6 7 8
2
运行结果
[1, 2], [1, 3], [2, 3], [4, 5], [4, 6], [5, 6]
  • 解析:
    • 输入数组 [1, 2, 3, 4, 5, 6, 7, 8] 被分为 link1 = [1, 2, 3]link2 = [4, 5, 6, 7, 8]
    • 输入 num = 2,表示需要生成长度为 2 的组合。
    • 最终结果为 link1link2 中长度为 2 的所有组合。

总结

  • 该代码通过分组和深度优先搜索(DFS),实现了根据输入条件生成指定长度的组合。
  • 适用于需要处理分组和组合问题的场景,尤其是需要根据条件动态生成结果的场景。

如果有其他问题,欢迎随时提问!


http://www.ppmy.cn/ops/149454.html

相关文章

Go oom分析(二)——导出dump离线分析

在 Go 程序中导出内存或 CPU 的 dump 文件&#xff08;通常通过 pprof 工具生成&#xff09;并进行分析&#xff0c;以下是详细步骤&#xff1a; 1. 在程序中开启 pprof 在你的 Go 程序中引入 net/http/pprof&#xff0c;开启 pprof 服务&#xff1a; import (_ "net/h…

利用AI大模型和Mermaid生成流程图

核心点1&#xff1a;利用大模型生成流程图的语句&#xff08;Code&#xff09; 确定业务流程&#xff1a; 用户需要明确要绘制的业务流程&#xff0c;包括主要步骤、决策点以及各步骤之间的关系。将确定的业务流程以文字形式描述出来。 生成Mermaid代码&#xff1a; 将描述好的…

Windows下调试Dify相关组件(2)--后端Api

1.部署依赖的服务&#xff08;代码最外层的docker目录&#xff09; 1.1 将middleware.env.example复制&#xff0c;并改名为middleware.env。 1.2 查看docker-compose.middleware.yaml&#xff0c;有5个服务 db&#xff1a;postgres数据库。 redis&#xff1a;redis缓存。 sa…

[ComfyUI]接入Google的Whisk,巨物融合玩法介绍

一、介紹​ 前段时间&#xff0c;谷歌推出了一个图像生成工具whisk&#xff0c;有一个很好玩的图片融合玩法&#xff0c;分别提供三张图片,就可以任何组合来生成图片。​ ​ 最近我发现有人开发了对应的ComfyUI插件&#xff0c;对whisk做了支持&#xff0c;就来体验了下&#…

2024年12月中国电子学会青少年软件编程(Python)等级考试试卷(二级)答案 + 解析

2024.12 中国电子学会青少年软件编程&#xff08;Python&#xff09;等级考试试卷&#xff08;二级&#xff09;答案 解析 一、单选题(共25题&#xff0c;共50分) 1. 已知字典如下 dic1 { name: Ming, age:20, grade: A, Tel:6666666 } 以下哪个代码运行结果为20&#xff…

多个表单使用相同的 ref 和 rules,表单验证规则不生效

在 Vue 和 Element UI 中&#xff0c;如果多个表单使用相同的 ref 和 rules&#xff0c;可能会导致表单验证规则不生效。这是因为 ref 是唯一的&#xff0c;多个表单共享同一个 ref 会导致冲突。 解决方法&#xff1a; 1. 为每个表单设置不同的 ref 为每个表单设置不同的 re…

VTK知识学习(29)-交互问题

1、窗体类型 1&#xff09;、RenderWindowControl UserControl derived implementation of vtkRenderWindow for use in Windows Forms applications. The client area of this UserControl is completely filled with an instance of a vtkRenderWindow. 用于Windows窗体应…

本地系统A与云平台B数据对接demo演示

.一、云平台B提供工程类库【.net工程类库&#xff0c;直接调用】 本地系统A-本地电脑调用&#xff0c;电脑连接互联网 云平台B-互联网云平台-提供cs文件&#xff0c;接口文档。 1.c#类库 2.控制台或winform直接调用c#类库&#xff1b; 3.云平台B输出【c#源文件】 4.cs文件端…