记录一次解题过程

记录一次解题过程

""“
有角色ID分别是1,2,3,4,5,6,要求同一用户不能同时拥有1,2,3,4中的任意两个或多个角色,5,6的无所谓, 要求给出快速判断的逻辑
 解决方案:
 1. 将角色ID转换为二进制表示:1 -> 001, 2 -> 010, 3 -> 100, 4 -> 1000, 5 -> 10000, 6 -> 1000002. 对于用户拥有的角色ID列表,我们可以将它们的二进制表示进行按位或(OR)运算,得到一个结果值。
 3.检查结果值的二进制表示中,后四位(对应角色ID 1, 2, 3, 4)是否有多个1。如果有多个1,说明用户拥有多个不允许的角色。
"""


def has_conflicting_roles(role_ids: list):
    role_binaries = {
        1: 0b001,
        2: 0b010,
        3: 0b100,
        4: 0b1000,
        5: 0b10000,
        6: 0b100000
    }

    result = 0
    for role_id in role_ids:
        result |= role_binaries[role_id]
        print(bin(result))
    # 检查后四位是否有多个1
    return bin(result)[-4:].count('1') > 1


if __name__ == '__main__':
    print(has_conflicting_roles([1, 2]))  # 输出 True
    print(has_conflicting_roles([1, 5]))  # 输出 False
    print(has_conflicting_roles([2, 3, 6]))  # 输出 True
    print(has_conflicting_roles([5, 6]))  # 输出 False

你可能感兴趣的:(python)