容斥原理模板

容斥原理

容斥原理：
n个集合
S = 1个集合 - 两个集合交集 + 三个集合交集 + … + (-1) ^ n * n个集合交集。总共有(2 ^ n) - 1项。

用一个韦恩图表示

给ABC面积，求图形总占面积
S = A + B + C - AB - AC - BC + ABC
当给四个集合时，
S = A + B + C + D - AB - AC - AD -BC - BD - CD + ABC + ABD + ACD + BCD - ABCD;
类推到n个集合：
S = 1个集合 - 两个集合交集 + 三个集合交集 + … + (-1) ^ n * n个集合交集。总共有(2 ^ n) - 1项。
证明：根据S的意义，一个集合 = Cn1，两个集合交集 = Cn2，以此类推。
由于Cn1 + Cn2 + Cn3 +…+Cnn = 2 ^ n,所以总共有(2 ^ n) - 1项。

来道模板题：

给定一个整数 n 和 m 个不同的质数 p1,p2,…,pm 。

请你求出 1∼n 中能被 p1,p2,…,pm 中的至少一个数整除的整数有多少个。

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

第二行包含 m 个质数。

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

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

题目意思为给一个n和m个质数，求1 - n中能被质数中其中一个整除的个数。如果暴力解必超时，复杂度为O（nm）。

所以可以对每个质数，求出1-n中能整除每个质数及其倍数的数量分个m个集合，再利用容斥原理计算各个集合相交在一起的总面积。
例如样例中，2的倍数的集合大小：10 / 2 = 5，3的倍数的集合大小：10 / 3 = 3，重叠部分集合大小：10 / 6 = 1，答案就为S2 + S3 - S23 = 7。
S = 1个集合 - 两个集合交集 + 三个集合交集 + … + (-1) ^ n * n个集合交集。

这里还有一个小技巧，利用位运算来模拟各个集合的相交情况，总共有(2 ^ m) - 1项，所以可以用二级制位数为m的整数来模拟各个集合的选取情况，哪几位为1就表示哪几个质数的集合。
见代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N = 20;

int n,m;
int a[N];

int main()
{
    cin>>n>>m;
    for(int i = 0; i < m; i ++ )
        cin>>a[i];

    int res = 0;
    //位运算模拟方案数
    for(int i = 1; i < 1<<m; i ++ )//1 << m = 2 的 m次方，转换为遍历所有种方案
    {//这里不能等于，比如当n = 2时，应该选的是两位，如果等于就可以选到第三位了
        int cnt = 0;//方案中选a数组中的数量
        int t = 1;
        for(int j = 0; j < m; j ++ )//j个位置选
            if(i >> j & 1)
            {
                cnt ++;
                if((ll) t * a[j] > n)//因为是多个质数相乘，可能大于n，忽略
                {
                    t = -1;
                    break ;
                }
                t = t * a[j];
            }
        if(t != -1)//奇数为加，偶数为减
        {
            if(cnt % 2)
                res += n / t;
            else
                res -= n / t;
        }
    }
    cout<<res;
    return 0;
}