hdu - 4335 - What is N? - 数论

数论啊。http://acm.hdu.edu.cn/showproblem.php?pid=4335



给定bpm，求有几个n满足条件。



证明见http://hi.baidu.com/aekdycoin/item/e493adc9a7c0870bad092fd9

题目给定b,p,M 问 0 <= n <= M 中有多少个数满足 n^(n!) ≡ b (MOD p)，并且可恶的是这里没有提到任何数的特殊性质，给定都只是限定在正整数且p>0。

有欧拉定理我们知道 n^(phi(p)) ≡ 1 (mod p) 但是这里要求gcd(n, p) = 1,显然题目并没有这么要的数据，那么如果题目给定是满足n,p互质的话，那么我们就可以知道

n^(x) mod p 是有循环节的，这个循环节就是n^(phi(p))，如果n,p不互质的话，那么我们可以证明这个循环节 T | phi(p)，所以我们还是可以选择phi(p)来作为循环节，于是

下面我们就可以分四部分来进行计算了。

part_1:首先计算n! < phi(p) 顶多8个点，for循环直接暴力

part_2:如果还剩余有区间的话，在计算 phi(p) <= n! < s!, s=min{x|x! mod phi(p)=0} // 此时的n!次方已经存在一个循环节，但是n还是变化的

part_3:接下来的区间就是n! % phi(p) = 0, 这说明其已经完全依赖于phi(p)的值了，所以只要考虑到n的循环性质了，直接统计n的一个循环节里面出现次数就可以了

part_4:把余下的n进行统计即可

#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define MAXN 100000
using namespace std;

typedef unsigned long long int Int64;

int p[MAXN+5], pm[10000], phi[MAXN+5], idx = -1, MOD, B;

int Fac[MAXN+5];

Int64 M, ans;

void GetPrime()
{
for (int i = 2; i <= MAXN; ++i) {
if (!p[i]) {
pm[++idx] = i;
}
for (int j = 0; pm[j]*i <= MAXN; ++j) {
p[pm[j]*i] = 1;
if (i % pm[j] == 0) {
break;
}
}
}
/*    printf("%d\n", idx);
for (int i = 0; i <= 100; ++i) {
printf("%d ", pm[i]);
}*/
}

void Eular()
{
phi[1] = 1;
for (int i = 2; i <= MAXN; ++i) {
//    printf("phi[%d] = %d\n", i-1, phi[i-1]);
//    getchar();
if (!p[i]) {
phi[i] = i - 1;
continue;
}
for (int j = 0; pm[j]*pm[j] <= i; ++j) {
if (i % pm[j] == 0) {
if (i / pm[j] % pm[j] == 0) {
phi[i] = pm[j] * phi[i/pm[j]];
}
else {
phi[i] = phi[pm[j]] * phi[i/pm[j]];
}
break;
}
}

}
}

bool _pow(Int64 a, Int64 b)
{
Int64 ret = 1;
while (b) {
if (b & 1) {
ret *= a;
ret %= MOD;
}
a *= a;
a %= MOD;
b >>= 1;
}
return ret == B;
}

void deal()
{
int LIM = -1, ptr;
for (ptr = 0; ptr <= M; ++ptr) {
// 第一段，小于phi[MOD]的一部分
if (!ptr) { Fac[ptr] = 1; }
else { Fac[ptr] = Fac[ptr-1] * ptr; }
if (Fac[ptr] >= phi[MOD]) {
break;
}
if (_pow(ptr, Fac[ptr])) {
++ans;
}
}
if (ptr > M) { return; } // 如果已经超过了M的话
for (int i = ptr; i <= M; ++i) {
if (!i) { Fac[i] = 1 % phi[MOD]; }
else { Fac[i] = (Fac[i-1] * i) % phi[MOD]; }
if (Fac[i] == 0) {  // 如果找到了这个点
LIM = i;
break;
}
if (_pow(i, Fac[i]+phi[MOD])) {
++ans;
}
}
if (LIM == -1) { return; } // 如果没有找到一个是phi[MOD]倍数的点
int t = 0;
if ((M - LIM+1) >= MOD) { // LIM 这个点是没有计算的，所以长度为M-LIM+1，这里判定剩余是否大于MOD
for (int i = LIM; i < LIM+MOD; ++i) {
if (_pow(i%MOD, phi[MOD])) { // 由于此时已经是phi[MOD]的倍数，所以直接把phi[MOD]这个指数代入
++t;
}
}
ans += (M - LIM+1) / (1LLU*MOD) * t;
}
int left = (M-LIM+1) % MOD; // 残余的余数
for (int i = LIM; i < LIM+left; ++i) {
if (_pow(i%MOD, phi[MOD])) {
++ans;
}
}
}

int main()
{
GetPrime();
Eular();
int T, ca = 0;
scanf("%d", &T);
while (T--) {
ans = 0;
scanf("%d %d %I64u", &B, &MOD, &M);
if(M == 18446744073709551615LLU && MOD == 1 && B == 0) {
// 由于b=0,p=1,所以可以直接得到M+1，由于溢出原因所以需要特判
printf("Case #%d: 18446744073709551616\n", ++ca);
}
else
{
deal();
printf("Case #%d: %I64u\n", ++ca, ans);
}
}
return 0;
}