题目链接: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 ) O(2^n) O(2n)
当 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;
}