分组飞行棋子| 豆包MarsCode AI刷题

72 阅读2分钟

问题描述

小M和小F在玩飞行棋。游戏结束后,他们需要将桌上的飞行棋棋子分组整理好。现在有 N 个棋子,每个棋子上有一个数字序号。小M的目标是将这些棋子分成 M 组,每组恰好5个,并且组内棋子的序号相同。小M希望知道是否可以按照这种方式对棋子进行分组。

例如,假设棋子序号为 [1, 2, 3, 4, 5],虽然只有5个棋子,但由于序号不同,因此不能形成有效的分组。如果序号是 [1, 1, 1, 1, 1, 2, 2, 2, 2, 2],则可以形成两个有效分组,因此输出为 True

测试样例

样例1:

输入:nums = [1, 2, 3, 4, 5]
输出:"False"

样例2:

输入:nums = [1, 1, 1, 1, 2, 1, 2, 2, 2, 2]
输出:"True"

样例3:

输入:nums = [5, 5, 5, 5, 5, 5, 5, 5, 5, 5]
输出:"True"

样例4:

输入:nums = [7, 7, 7, 8, 8, 8, 8, 8, 7, 7]
输出:"True"

样例5:

输入:nums = [9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9]
输出:"False"

我的python版本:

from typing import Counterdef solution(nums):    p = Counter()    for k in nums:        p[k] += 1        for k in p:        if(p[k] % 5 != 0):            return "False"    return "True"if __name__ == "__main__":    #  You can add more test cases here    print(solution([1, 3, 4, 5, 6, 5, 4]) == False )    print(solution([1, 1, 1, 1, 2, 1, 2, 2, 2, 2]) == True)    print(solution([11, 45, 49, 37, 45, 38, 3, 47, 35, 49, 26, 16, 24, 4, 45, 39, 28, 26, 14, 22, 4, 49, 18, 4, 4, 26, 47, 14, 1, 21, 9, 26, 17, 12, 44, 28, 24, 24, 10, 31, 33, 32, 23, 41, 41, 19, 17, 24, 28, 46, 28, 4, 18, 23, 48, 45, 7, 21, 12, 40, 2, 19, 19, 28, 32, 6, 27, 43, 6, 18, 8, 27, 9, 6, 6, 31, 37, 15, 26, 20, 43, 3, 14, 40, 20]) == False)

算法步骤

  1. 统计每个序号的棋子数量:遍历输入的棋子序列,使用 Counter 统计每个序号的棋子数量。
  2. 检查每个序号的棋子数量是否能被5整除:遍历 Counter 中的每个序号,检查其对应的棋子数量是否能被5整除。如果有任何一个序号的棋子数量不能被5整除,则返回 False
  3. 返回结果:如果所有序号的棋子数量都能被5整除,则返回 True

代码实现思路

  1. 初始化 **Counter**:创建一个空的 Counter 对象。
  2. 统计棋子数量:遍历输入的棋子序列,更新 Counter 中的计数。
  3. 检查每个序号的棋子数量:遍历 Counter 中的每个序号,检查其数量是否能被5整除。
  4. 返回结果:根据检查结果返回 TrueFalse

这就是今天份的总结啦