牛客网暑期ACM多校训练营（第三场） D.Big Intege良心题解

D.Big Intege

题意

已知$A(n)=11111\cdots 11111$(n为1的数量)
求$A(i^j)\mathrm{m}\mathrm{o}\mathrm{d}p=0$的数量$(1\le i\le n,1\le j\le m)$

思路

公式推导

第一步

令$n=i^j$
显然，$11111\cdots 11111$是个很优美的数字
$A(n)=\frac{10^n-1}{9}$

第二步

$$\frac{10^n-1}{9}\%p==0$$
除了$p==3$,$p$和$9$都是互质的，那么

• $\frac{10^n-1}{9}\%p$
• $=(10^n-1)\ast inv9\%p$
• $=((10^n-1)\%p)\ast (inv9\%p)\%p$
• $=0$

第三步

因为互质
$inv9̸=0$
$((10^n-1)\%p)\ast (inv9\%p)\%p$
所以推出$$10^n-1\%p==0$$$$10^n\equiv 1\mathrm{m}\mathrm{o}\mathrm{d}p$$

第四步

用以下随便哪个定理都一样
费马小定理:
**若$a$不是$p$的倍数，$a^{p-1}\equiv 1\mathrm{m}\mathrm{o}\mathrm{d}n$ **
显然$p-1$属于$n$,但$p-1$不一定最小的满足的数
若存在$d(d\le p-1)$满足上式，$d$一定是$p-1$的约数（同余式的性质）

欧拉定理引理：
若$a,n$互质，则满足$a^x\equiv 1\mathrm{m}\mathrm{o}\mathrm{d}n$的最小正整数$x$是$\Theta (n)$的约数

如此枚举约数即可求出最小的x满足$10^x\equiv \mathrm{m}\mathrm{o}\mathrm{d}1$

int big_d[32000];
LL Get_Min_d(LL p) {
	LL phi = p - 1; int k = 0;
	for (int i = 1; i * i <= phi; i++) {
		if (phi % i == 0) {
			if (quickpow(10, i, p) == 1)	//快速幂
				return i;
			big_d[++k] = phi / i;	//大的因数先存起来
		}
	}
	for (int i = k; i >= 1; i--) {		//小的因数找不到，再找大的
		if (quickpow(10, big_d[i], p) == 1)
			return big_d[i];
	}
}

第五步

显然满足$10^d\mathrm{m}\mathrm{o}\mathrm{d}p==1$的最小循环节d只有一个
证明：
若存在$d_1,d_2$且$d_1$和$d_2$互质
$10^{d_1}\mathrm{m}\mathrm{o}\mathrm{d}p==1$
$10^{d_2}\mathrm{m}\mathrm{o}\mathrm{d}p==1$
那么必定存在$10^{(d_2-d_1)}\mathrm{m}\mathrm{o}\mathrm{d}p==1$,依次类推做辗转相减法
存在$d=gcd(d_1,d_2)$,那么d就是唯一循环节

问题转化

若我们找到最小循环节d，那么所有满足$10^d\mathrm{m}\mathrm{o}\mathrm{d}p==1$,那么只要找到所有$i^j$中的倍数即可

$i^j$中d的倍数

ps：此处公式中$[x]$为将x向上取整

第一步

将d分解质因数，$d=p_1^{a_1}\ast p_2^{a_2}\ast p_3^{a_3}\ast \cdots \ast p_n^{a_n}$

第二步

若确定$j$,则$i$一定是min_d的倍数
$min(d)=p_1^{[\frac{a_1}{j}]}\ast p_2^{[\frac{a_2}{j}]}\ast p_3^{[\frac{a_3}{j}]}\ast \cdots \ast p_n^{[\frac{a_n}{j}]}$

因为
$$min(d{)}^j=p_1^{[\frac{a_1}{j}]\ast j}\ast p_2^{[\frac{a_2}{j}]\ast j}\ast p_3^{[\frac{a_3}{j}]\ast j}\ast \cdots \ast p_n^{[\frac{a_n}{j}]\ast j}$$
$$=d\ast p_1^{[\frac{a_1}{j}]\ast j-a_1}\ast p_2^{[\frac{a_2}{j}]\ast j-a_2}\ast p_3^{[\frac{a_3}{j}]\ast j-a_3}\ast \cdots \ast p_n^{[\frac{a_n}{j}]\ast j-a_n}$$

例如：对于$12=2^2\ast 3^1$
当$j=1$,$min(d)=2^2\ast 3^1=12$,$12$的倍数的$1$次方,为$12$的倍数
当$j=2$,$min(d)=2^1\ast 3^1=6$,$6$的倍数的$2$次方,为$12$的倍数
当$j=3$,$min(d)=2^1\ast 3^1=6$,$6$的倍数的$3$次方,为$12$的倍数

第三步

显然$1-n$中min_d的倍数个数为$n/min(d)$
对于$1\le j\le max(a_1,a_2,\cdots ,a_n)$,$min(d)$是不同的
对于$max(a_1,a_2,\cdots ,a_n)<j$,$min(d)=p_1\ast p_2\ast p_3\ast \cdots \ast p_n$
先分解质因数
再枚举$j$计算，min_d的倍数个数

分解质因数

int p[32], a[32], cnt;	//p数组为质因数，a数组为次数
void Dec(LL d) {
	fill(p, p + 32, 0);
	fill(a, a + 32, 0); cnt = 0;
	for (int i = 2; i * i <= d; i++) {
		if (d % i == 0) {
			p[++cnt] = i;
			while (d % i == 0) {
				d /= i;
				a[cnt]++;
			}
			if (d == 1)return;
		}
	}
	if (d > 1) p[++cnt] = d, a[cnt] = 1;
	return;
}

枚举j，计算d的倍数个数

LL p_power[32][32];
LL Cal(LL n, LL m) {
	for (int i = 1; i <= cnt; i++) {		//预处理，所有p[i]^a[i]
		p_power[i][1] = p[i];
		for (int j = 2; j <= a[i]; j++)
			p_power[i][j] = p_power[i][j - 1] * p[i];
	}

	int limit = 0;			//1-max(a[1]----a[n])的min_d都要单独计算
	for (int i = 1; i <= cnt; i++)limit = max(limit, a[i]);
	limit = min(LL(limit), m);

	int min_d = 1; LL ans = 0;
	for (int i = 1; i <= limit; i++) {		//枚举j计算
		min_d = 1;
		for (int j = 1; j <= cnt; j++)
			min_d *= p_power[j][ceil(a[j], i)];
		ans += n / min_d;
	}
	if (m > limit)ans += n / min_d * (m - limit);
	return ans;
}

AC代码

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#pragma warning (disable:4996)
typedef long long LL;
#define ceil(a,b) a % b == 0 ? a/b : a/b+1
LL quickpow(LL m, LL p, LL mod) {
	LL res = 1;
	while (p) {
		if (p & 1)
			res = res * m % mod;
		m = m * m % mod;
		p >>= 1;
	}
	return res;
}
int big_d[32000];
LL Get_Min_d(LL p) {
	LL phi = p - 1; int k = 0;
	for (int i = 1; i * i <= phi; i++) {
		if (phi % i == 0) {
			if (quickpow(10, i, p) == 1)
				return i;
			big_d[++k] = phi / i;
		}
	}
	for (int i = k; i >= 1; i--) {
		if (quickpow(10, big_d[i], p) == 1)
			return big_d[i];
	}
}
int p[32], a[32], cnt;
void Dec(LL d) {
	fill(p, p + 32, 0);
	fill(a, a + 32, 0); cnt = 0;
	for (int i = 2; i * i <= d; i++) {
		if (d % i == 0) {
			p[++cnt] = i;
			while (d % i == 0) {
				d /= i;
				a[cnt]++;
			}
			if (d == 1)return;
		}
	}
	if (d > 1) p[++cnt] = d, a[cnt] = 1;
	return;
}
LL p_power[32][32];
LL Cal(LL n, LL m) {
	for (int i = 1; i <= cnt; i++) {
		p_power[i][1] = p[i];
		for (int j = 2; j <= a[i]; j++)
			p_power[i][j] = p_power[i][j - 1] * p[i];
	}

	int limit = 0;
	for (int i = 1; i <= cnt; i++)limit = max(limit, a[i]);
	limit = min(LL(limit), m);

	int min_d = 1; LL ans = 0;
	for (int i = 1; i <= limit; i++) {
		min_d = 1;
		for (int j = 1; j <= cnt; j++)
			min_d *= p_power[j][ceil(a[j], i)];
		ans += n / min_d;
	}
	if (m > limit)ans += n / min_d * (m - limit);
	return ans;
}
int main() {
	int  t; scanf("%d", &t);
	LL p, n, m;
	while (t--) {
		scanf("%lld%lld%lld", &p, &n, &m);
		if (p == 2 || p == 5) {
			printf("0\n");
			continue;
		}
		else if (p == 3){
			printf("%lld\n", n / 3 * m);
			continue;
		}
		LL d = Get_Min_d(p);
		Dec(d);
		LL ans = Cal(n, m);
		printf("%lld\n", ans);
	}
}