6272. 2019.8.4【NOIP提高组A】整除 (division) （中国剩余定理 ， 积性筛）

6272. 2019.8.4【NOIP提高组A】整除 (division) 
(File IO): input:division.in output:division.out

Time Limits: 3000 ms  Memory Limits: 262144 KB  Detailed Limits  

Description

Input

Output

Sample Input

0
1
2 3
2 3

Sample Output

6

Data Constraint

 

题目大意：

给你整数n 和 m ， 求[1,n]内有几个x满足 (x^m - x) % n ==0

 

引理1：

高次同余方程的解数及解法定理1 ：

若是k个两两互质的正整数，，则同余式

      

                                    (1)

 

与同余式组

      

           (i=1,2,3,...,k)          (2)

 

等价，并且若用表示对模的解数，T表示(1)式对模m的解数，

 

则：

 

题解 ： 

 首先 ， 一个同余方程的解x ,x <= mod数 。大于模数的解是没有意义的（因为这样就有无数个了）

对于题目输入的n的质因子 。 求出[1,p]范围内x满足(x^m - x) % p ==0 的个数

答案即他们的乘积。

x^m次方不能暴力 ， 要O（n）预处理 出每个数的m次方。

只用我们的素数线筛加点东西就行了。

我写的线筛blog ： https://blog.csdn.net/Com_man_der/article/details/95107680

mcf[i]表示i的m次方是多少 ,i是质数 ， 直接快速幂 。其他的 ， 我们在线筛的时候 ， mcf[pri[j] * i] = mcf[pri[j]] * mcf[i]

具体见代码 ：

#include<bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
#define mcy(a,b) memcpy(a,b,sizeof(a))
#define mod 998244353
#define N 10010
#define open(x) freopen(x".in","r",stdin);freopen(x".out","w",stdout)
using namespace std;

int id,T,c,m,i,o,cnt;
int  pri[N] , mcf[N];
long long ans;

int power(int x,int p,int MOD)
{
	int ans = 1;
	while(p)
	{
		if(p&1)(ans*=x)%=MOD;
		(x*=x)%=MOD;
		p>>=1;
	}
	return ans;
}

int check(int x,int p)
{return (mcf[x] - x + p)%p==0; }

int minp[N],len=0;

void ol(int p )
{
	mem(minp,0);mem(mcf,0);
	len=0;
	mcf[1] = 1 % p;
	for(int i=2;i<=p;i++)
	{
		if(minp[i] == 0 )
		{
			pri[++len] = i;
			minp[i] = i;
			mcf[i] = power(i , m , p);
		}
		for(int j=1;j<=len;j++)
		{
			int  k=pri[j] * i;
			if(k>p || pri[j] > minp[i]) break;
			minp[k]=pri[j];
			mcf[k] = mcf[pri[j]] * mcf[i] % p ;
		}
	}
}

int sol(int p)
{
	ol(p);
	cnt=0;
	for(int i=1;i<=p;i++) if(check(i,p)) ++cnt;
	return cnt % mod;
}

int main()
{
	open("division");
	scanf("%d" , &id);
	scanf("%d",&T);
	while(T--)
	{
		ans=1ll;
		scanf("%d%d",&c,&m);
		for(i=1;i<=c;i++) 
		{
			scanf("%d",&o);
			(ans *= sol(o))%=mod;
		}
		printf("%lld\n",ans);
	}
}

O(Tct)两千五百万次 ， 注意常数。