AI 刷题 105 | 豆包MarsCode AI刷题

问题描述

某特种部队采用了一套性格密码机制来筛选执行特定任务的最佳士兵，该机制的规则如下：

1. 每个人的性格可以通过 M 个维度来描述，每个维度分为 A, B, C, D, E 五种类型。
2. 同一维度内，字母距离越近，性格类型差异越小，匹配程度越高。比如，A 和 B 的差异为 1，A 和 D 的差异为 3。
3. 其中 AE、BD、CE、BE 为不相容性格类型，差异值设为无穷大（无法匹配）。
4. 如果某一维度存在不相容性格类型，则表示两个士兵性格完全不匹配。
5. 对于符合匹配条件的士兵，差异值总和越小表示匹配程度越高。

现在，有一个重要的机密任务，要求找到最匹配该任务所需性格密码的士兵。你需要编写一个算法，帮助部队找到符合条件的最佳人选。

• m 表示性格密码的维度。
• n 表示备选特种兵的数量。
• target 是代表任务的性格密码。
• array 是一个包含 n 个元素的列表，每个元素为 M 位的性格密码。

测试样例

样例1：

输入：m = 6, n = 3, target = "ABCDEA", array = ["AAAAAA", "BBBBBB", "ABDDEB"]
输出：'ABDDEB'

样例2：

输入：m = 5, n = 4, target = "ABCED", array = ["ABCDE", "BCDEA", "ABDCE", "EDCBA"]
输出：'ABCDE'

样例3：

输入：m = 4, n = 4, target = "AEBC", array = ["ACDC", "BBDC", "EBCB", "BBBB"]
输出：'None'

问题分析

设计一个算法来解决“性格密码匹配”的问题。目标是找到最符合任务性格密码的士兵，其性格密码与任务目标的差异值最小。如果某士兵的某维度存在不相容性格类型，则认为该士兵不符合条件。具体规则如下：

1. 输入数据：

   • m：性格密码的维度数。
   • n：备选士兵的数量。
   • target：目标任务的性格密码。
   • array：长度为 n 的列表，每个元素是一个 m 位的性格密码。

2. 差异计算规则：

   • 性格类型为 A, B, C, D, E，按顺序排列，差异为字母间的距离：

     • 差异值：例如A和B的差异为1，A和D的差异为3。

   • 不相容性格类型：

     • AE, BD, CE, BE 的差异值设为无穷大。

3. 匹配条件：

   • 如果某士兵的某维度与目标密码存在不相容性格类型，则该士兵直接被排除。
   • 在所有符合条件的士兵中，选出差异值总和最小的士兵。

4. 输出：

   • 如果找到最匹配的士兵，返回其性格密码和差异值。
   • 如果没有符合条件的士兵，返回 "None"。

解决思路

1. 预处理差异值：

   • 构建一个差异值查找表，将所有字母对的差异值预先计算好，包括不相容类型的特殊处理（设为无穷大）。

2. 逐个比较：

   • 遍历每个士兵的性格密码，逐维度与target比较差异。
   • 如果某维度差异为无穷大，则直接排除该士兵。
   • 计算差异值总和，并记录当前最优解。

3. 输出结果：

   • 如果有符合条件的士兵，输出差异值最小的士兵及其总差异值。
   • 如果没有符合条件的士兵，输出 "None"。

主要代码如下：

复杂度分析

• 时间复杂度： O(n⋅m+n⋅log⁡n)

  • 计算差异值的时间复杂度：O(n⋅m)
  • 排序操作的时间复杂度：O(n⋅log⁡n)

• 空间复杂度：O(n⋅m)

  • 存储符合条件的候选士兵和每个士兵的差异值。