记录一次解题过程
""“
有角色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 -> 100000。
2. 对于用户拥有的角色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