题解 | 计数问题

计数问题

https://www.nowcoder.com/practice/ca91bb01fb1b4d20a1968dddc9884c1c

import sys
import math
from math import gcd

# 计算两个数的最小公倍数
def lcm(a, b):
    return a * b // gcd(a, b)


# 计算一个列表所有数的最小公倍数
def multi_lcm(nums):
    res = 1
    for num in nums:
        res = lcm(res, num)
    return res


def solve():
    data = sys.stdin.read().split()
    idx = 0
    T = int(data[idx])
    idx += 1
    for _ in range(T):
        n, L, R = map(int, data[idx : idx + 3])
        idx += 3
        a = list(map(int, data[idx : idx + n]))
        idx += n
        a = list(set(a))  # 去重,减少子集数量
        n = len(a)
        ans = 0
        # 枚举所有非空子集,用二进制位表示:1~(1<<n)-1
        for mask in range(1, 1 << n):
            subset = []
            cnt = 0  # 子集的元素个数
            for i in range(n):
                if mask & (1 << i):
                    subset.append(a[i])
                    cnt += 1
            # 计算子集的最小公倍数
            m = multi_lcm(subset)
            if m > R:
                continue  # 无满足条件的数,剪枝
            # 区间内能被m整除的数的个数
            num = R // m - (L - 1) // m
            # 容斥:奇数加,偶数减
            if cnt % 2 == 1:
                ans += num
            else:
                ans -= num
        print(ans)


if __name__ == "__main__":
    solve()

全部评论

相关推荐

2本硕,在这一个下午真的绷不住了,浪费了太多时间,现在的技术栈还停在C语言和stm32上,找嵌入式的实习面试被拷打,找杭州的一个也找不到,真的心里难受,linux没学过,研二了开始慌了。
一条淡水魚:嵌入式这行的面试我认为实际项目比较重要,技术栈简单的提一嘴就行,面试官在乎的关键点在于你用了这些技术做了哪些工作解决了什么问题,而不是停留在离散的那些个技术栈上,那除了教课没有意义,好比你提到的c语言和32,你用32做过哪些具体的项目?接触过什么外设?使用过哪些公司的SDK?有没有实际产品落地?以及各种只有进入真正的生产环节当中才会积累到的经验......主动去和面试官讨论这些实际的问题,甚至还能就某个具体参数的合理性与他去简单探讨一下,只要技术栈对口,基本上就稳啦~(另外linux和RTOS是嵌入式的标配哦,选一个方向走下去吧)
点赞 评论 收藏
分享
不知道怎么取名字_:玩游戏都写到简历上了啊
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务