集合论和位运算 LeetCode 第393周赛第三题

题解

class Solution {
public:
    long long findKthSmallest(vector<int>& coins, int k) {
        auto check = [&](long long m) -> bool {
            long long cnt = 0;
            for (int i = 1; i < (1 << coins.size()); i ++) {
                // 二进制枚举每种情况  如果是三个集合  A B C 那么此处枚举 A B C AB AC BC ABC  用位表示
                long long lcm_res = 1; // 计算子集的LCM
                for (int j = 0; j < coins.size(); j ++) {
                    if (i >> j & 1) {
                        // 如果 j 在集合 i 中，求lcm
                        lcm_res = lcm(lcm_res, coins[j]);
                        if (lcm_res > m) break;
                    }
                }
                // 此处计算的lcm是整个子集的情况，此时计算 m / lcm_res 就是集合数字个数
                // __builtin_popcount(i) 求i集合这种情况包含了多少种集合，是奇数个还是偶数个
                cnt += __builtin_popcount(i) % 2 ? m / lcm_res : -m / lcm_res;
            }
            return cnt >= k;
        };

        long long left = k - 1, right = (long long) ranges::min(coins) * k;
        while (left + 1 < right) {
            long long mid = (left + right) / 2;
            (check(mid) ? right : left) = mid;
        }
        return right;
    }
};

本题我认为对我来说醍醐灌顶的是用二进制来枚举集合的每一种情况

比如我现在有A B C三个集合，我想要求他们的并集

$A\cup B\cup C$ = $A+B+C-A\cap B-A\cap C-B\cap C+A\cap B\cap C$

那么当我们想要求表达式左边的时候，我们需要表达式右边每一项的值。

此时我们可以使用三位二进制位来表达每一项，
比如

$001->A$

$011->A\cap B$

以此类推就可以枚举出计算所需要的全部情况了
然后我们通过判断当前这个情况有哪些集合（此题就是数字）被选中了，进行相关操作（此题是求这些被选中的数字的最小公倍数以用来求可以生成的数字个数）

总结

最重要的是用二进制来表达信息的操作，这个操作和Linux的一些处理很像，比如epoll的一些状态的设置，使用不同的二进制位表达不同的选择，然后可以使用|（或操作）来设定初始的状态。

更直接的，某块虚拟内存块的可读可写 ，使用 WRITE | READ 来表达，这两个宏的背后也是二进制数

参考

题目本体 https://leetcode.cn/problems/kth-smallest-amount-with-single-denomination-combination/
位运算 https://leetcode.cn/circle/discuss/CaOJ45/