第四章 数学知识 容斥原理

1、算法思路

在组合数学里，集合A1、A2…An均为有限集，则这些集合的元素之和
三个集合的情况如下图所示：
$$\bigcup _1^{3}Ai=|A1|+|A2|+|A3|-|A1\cap A2|-|A1\cap A3|-|A2\cap A3|+|A1\cap A2\cap A3|$$

2、例题能整除的数

给定一个整数 n 和 m 个不同的质数 p1,p2,…,pm。
请你求出 1∼n 中能被 p1,p2,…,pm 中的至少一个数整除的整数有多少个。

输入格式
第一行包含整数 n 和 m。
第二行包含 m 个质数。

输出格式
输出一个整数，表示满足条件的整数的个数。

数据范围
1≤m≤16,
1≤n,pi≤10^9
输入样例：
10 2
2 3
输出样例：
7

#include<iostream>
using namespace std;
const int N = 20;
int p[N];

int main()
{
    int n,m;
    cin >> n >> m;
    for(int i = 0; i < m; i ++) cin >> p[i];
    int res = 0;
    for(int i = 1; i < 1 << m; i ++)
    {
        int t = 1, r = 0;
        for(int j = 0; j < m; j ++)
        {
            if(i >> j & 1)
            {
                if((long long)t * p[j] > n) 
                {
                    t = -1;
                    break;
                }
                r ++;
                t *= p[j];
            }
        }
        if(t == -1) continue;
         if(r % 2) res += n / t;
            else res -= n / t;
    }
    cout << res;
}

1. 举个例子
   p={2,3} n = 10
   能整除2的元素集合为S1={2,4,6,8,1}
   能整除3的元素集合为S2={3,6,9}
   能
   |S1∪S2| = |S1| + |S2| - |S1∩S2|,
2. 整个计算一共需要枚举$C_n^1+C_n^2+C_n^3+...+C_n^n=2^n-1$项，每一项的正负号是由每一项里的元素个数决定的。当个数为奇数的时候符号位为+，当个数为偶数的时候符号位为-。所以这里我们用二进制的数1<<m进行枚举，用r来记录m中1的个数，这个个数就是每一项中元素的个数。
3. 每一项元素的个数是这样计算的。对于上例中，|S1|=10 / 2 = 5,|S2| = 10 / 3 = 3,|S1∩S2| = 10 / 2 * 3 = 1，所以总数位7个

参考资料

Acwing