题解 | 计数问题

计数问题

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()

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
正在热议
更多
# 春招至今,你的战绩如何? #
5154次浏览 48人参与
# 你的实习产出是真实的还是包装的? #
1114次浏览 27人参与
# 米连集团26产品管培生项目 #
4209次浏览 198人参与
# 军工所铁饭碗 vs 互联网高薪资,你会选谁 #
6921次浏览 37人参与
# 简历第一个项目做什么 #
31257次浏览 313人参与
# 当下环境,你会继续卷互联网,还是看其他行业机会 #
186369次浏览 1115人参与
# MiniMax求职进展汇总 #
22940次浏览 297人参与
# 面试紧张时你会有什么表现? #
30380次浏览 188人参与
# 简历中的项目经历要怎么写? #
309398次浏览 4154人参与
# 网易游戏笔试 #
6324次浏览 83人参与
# 职能管理面试记录 #
10696次浏览 59人参与
# 把自己当AI,现在最消耗你token的问题是什么? #
6862次浏览 154人参与
# 从哪些方向判断这个offer值不值得去? #
56698次浏览 357人参与
# 腾讯音乐求职进展汇总 #
160398次浏览 1105人参与
# 小红书求职进展汇总 #
226849次浏览 1356人参与
# AI时代,哪些岗位最容易被淘汰 #
62445次浏览 731人参与
# 你怎么看待AI面试 #
179285次浏览 1165人参与
# 正在春招的你,也参与了去年秋招吗? #
362545次浏览 2631人参与
# 你的房租占工资的比例是多少? #
92125次浏览 896人参与
# 机械求职避坑tips #
94398次浏览 567人参与
# 校招笔试 #
466463次浏览 2951人参与
# 面试官最爱问的 AI 问题是...... #
27134次浏览 834人参与
牛客网
牛客网在线编程
牛客网题解
牛客企业服务