质数(prime)

题目大意

给出 $T$ 组询问，每组询问形如 $n,K$。

求满足下面条件的 $i$ 的个数：

1. $1\le i\le n$。
2. ${\forall }_{1\le jleqK},i\mathrm{m}\mathrm{o}\mathrm{d}{p}_j\ne 0$。

$p_j$ 表示第 $j$ 大的质数。

对于 $100\%$ 的数据，$1\le n\le 10^{18},1\le K\le 16$。

解题思路

对于 $K\le 7$ 的情况，考虑容斥，时间复杂度 $\mathcal{O}(2^K)$。

对于 $K>7$ 的情况，记 $cn{t}_i$ 表示 $[1,i]$ 中的一个数 $x$ 满足 ${\forall }_{1\le j\le 7},x\mathrm{m}\mathrm{o}\mathrm{d}{p}_j\ne 0$ 的数的个数。

因为，若一个数 $x$ 不是 $\forall 1\le i\le p,a_i$ 的倍数，当且仅当 $x\mathrm{m}\mathrm{o}\mathrm{d}Lcm(a)$ 不是 $\forall 1\le i\le p,a_i$ 的倍数。

反之，若一个数 $x$ 不是 $\forall 1\le i\le p,a_i$ 的倍数，则 $kx$ 不是 $\forall 1\le i\le p,a_i$ 的倍数，$k\ne 0$。

$n的ans=\lfloor n/510510\rfloor \ast cnt[510510-1]+cnt[n\mathrm{m}\mathrm{o}\mathrm{d}510510]$。

接下来用 $cnt$ 容斥即可，这个容斥是建立在 $[1,n]$ 中的数都不是前 $7$ 个质数的基础上进行容斥的，时间复杂度 $\mathcal{O}(2^{K-7})$。

CODE

#include <bits/stdc++.h>

using namespace std;

#define int unsigned long long

inline int read()
{
	int x = 0, f = 1;
	char c = getchar();
	while(c < '0' || c > '9')
	{
		if(c == '-') f = -1;
		c = getchar();
	}
	while(c >= '0' && c <= '9')
	{
		x = x * 10 + c - '0';
		c = getchar();
	}
	return x * f;
}

inline void write(int x)
{
	if(x < 0)
	{
		putchar('-');
		x = -x;
	}
	if(x > 9) write(x / 10);
	putchar(x % 10 + '0');
}

int T, n, k;

int cnt[510517], cnta[1 << 7 | 1], cntb[1 << 9 | 1];

const int kkk = 510510;

int prime[] = {0, 2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59};

int ans;

void init()
{
	int S = 1 << 7;
	for(int i = 0; i < S; ++i)
	{
		cnta[i] = 1;
		for(int j = 1; j <= 7; ++j)
			if(i & (1 << (j - 1)))
				cnta[i] *= prime[j];
	}
	for(int i = 1; i <= 7; ++i)
		for(int j = 1; j * prime[i] <= kkk; ++j)
			cnt[j * prime[i]] = 1;
	for(int i = 1; i < kkk; ++i)
		cnt[i] = cnt[i - 1] + (1 - cnt[i]);
	S = 1 << 9;
	for(int i = 0; i < S; ++i)
	{
		cntb[i] = 1;
		for(int j = 1; j <= 9; ++j)
			if(i & (1 << (j - 1)))
				cntb[i] *= prime[j + 7];
	}

}

int __cnt(int x)
{
	int k = 0;
	for(int i = x; i; i -= (i & -i))
		k++;
	return k;
}

signed main()
{
	init();
	T = read();
	while(T--)
	{
		n = read(), k = read();
		ans = 0;
		if(k <= 7)
		{
			int S = 1 << k;
			for(int i = 0; i < S; ++i)
			{
				if(__cnt(i) & 1)
					ans -= n / cnta[i];
				else
					ans += n / cnta[i];
			}
		}
		else
		{
			int S = 1 << (k - 7);
			for(int i = 0; i < S; ++i)
			{
				int p = n / cntb[i];
				int now = (p / kkk) * cnt[kkk - 1] + cnt[p % kkk];
				if(__cnt(i) & 1)
					ans -= now;
				else
					ans += now;
			}
		}
		write(ans);
		putchar('\n');
	}
	return 0;
}