容斥原理模板

题目链接：https://ac.nowcoder.com/acm/problem/14686
来源：牛客网

集合中的质数

题目描述

给出一个集合和一个数m。 集合里面有n个质数。 请你求出从 1 到 m 的所有数中，至少能被集合中的一个数整除的数的个数。

输入描述:

第一行两个正整数 n 和 m 。第二行n个正整数，分别为集合中的质数。

输出描述:

输出一个整数，表示符合要求的正整数的个数。

输入

3 37
5 7 13

输出

13

备注:

对于100%的数据，有n<=20，m为有符号64位正整数，集合内质数<=1000000000

n 个数的容斥原理，因为 n 只有 20，所以 dfs 足以解决。只需要进行一次枚举，就能枚举完所有的情况，时间复杂度为 $O(2^n)$
当 state 大于 m 之后，则 m / state = 0，之后再进行枚举就没有意义，可以进行剪枝
剪枝前运行时间为 19ms，剪枝后为 5ms

#include <iostream>
using namespace std;
typedef long long ll;

ll a[25];
ll n, m, ans;
// u表示安排了多少个数，当前的状态，选了多少个数
void dfs(ll u, ll state, ll cnt)
{
	// 安排完n个数
	if (u == n) {
		// 选了0个数
        if (cnt == 0) { 
            return;
        // 至少选了一个数
        } else { 
            if (cnt % 2 == 1) {
                ans += m / state;
            } else {
                ans -= m / state;
            }
        }
		return;
	}
	// 剪枝
	if (state > m) {
		return;
	}
	// 选
	dfs(u + 1, state * a[u], cnt + 1); 
	// 不选
	dfs(u + 1, state, cnt);
	
	return; 
}

int main(void)
{
	cin >> n >> m;
	for (int i = 0; i < n; i++)
		cin >> a[i];
	
	dfs(0, 1, 0);
	cout << ans << endl;
	 
	return 0;
} 

除了上面的 dfs 枚举，还可以利用二进制来进行枚举
运行时间66ms

#include <iostream>
using namespace std;
typedef long long ll;

ll a[25];

int main(void)
{
	ll n, m, res = 0;
	cin >> n >> m;
	for (int i = 0; i < n; i++)
		cin >> a[i];
	for (int i = 1; i < (1 << n); i++) {//用二进制来表示每个数的情况 
		int num = 0;//选了几个数 
		for (int j = i; j != 0; j >>= 1)
			num += j & 1;
		ll state = 1;
		for (int j = 0; j < n; j++) {
			if (i >> j & 1){
				state *= a[j];
				if (state > m)//剪枝 
					break;
			}
		}
		if (num % 2 == 0) 
			res -= m / state;
		else 
			res += m / state;
	}
	cout << res << endl;
	
	return 0;
}