【51nod 1038】X^A Mod P

題目描述

X^A mod P = B，其中P爲質數。給出P和A B，求< P的全部X。 例如：P = 11，A = 3，B = 5。 3^3 Mod 11 = 5 全部數據中，解的數量不超過Sqrt(P)。

分析

這道題包括幾個知識點

離散對數（大步小步BSGS算法）

求關於x的同餘方程$y^x \equiv n \pmod{P}（P爲質數）$的解， 設$m=\lceil \sqrt{n} \rceil,x=bm+r$，咱們預處理出$y^i(i\in[0,P-1])$，用map或hash儲存起來 咱們從小到大枚舉b，就能夠根據$y^{(b+1)m} \cdot n^{-1} \equiv y^{m-r}\pmod{P}$，在有序表中找到$y^{m-r}$來的到指數r

N次剩餘

求關於x的同餘方程$x^y \equiv n \pmod{P}（P爲質數）$的解 設P的原根爲g，由於$\varphi(P)=P-1$，根據原根的性質${1,2,3...P-1}$一一與${g^1,g^2,g^3...g^{P-1}}$對應 令$x=g^s，n=g^t$ 經過BSGS，咱們能夠求出t， 因而$s \cdot y \equiv t \pmod{P-1}$，用擴展歐幾里得解方程。

#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
#include <bitset>
#include <set>
#include <vector>
const int inf=2147483647;
const int mo=1e6+7;
const int N=100005;
using namespace std;
struct arr{  
    long long x,id;  
    bool operator<(const arr &b)const{  
        if (x == b.x) return id < b.id;  
        return x<b.x;  
    }  
}E[100500]; 
int T;
long long sol[N],P,A,B,G,m,num,ans[N],d[N];
long long ksm(long long x,long long y)
{
	long long s=1;
	for(;y;y>>=1,x=x*x%P)
		if(y&1) s=s*x%P;
	return s;
}
bool check(int G)
{
	int tmp=P-1;
	for(int i=2;i*i<=tmp;i++)
	{
		if(tmp%i) continue;
		if(ksm(G,i)==1 || ksm(G,tmp/i)==1) return false;
	}
	return true;
}
void pre()
{
	for(G=2;!check(G);G++);
	m=ceil(sqrt(P));
	long long v=1;
	for(int i=0;i<=m;i++)
	{
		arr tmp;
		tmp.x=v,tmp.id=i;
		E[i]=tmp;
		v=v*G%P;
	}
	sort(E,E+m);
}
long long BSGS(long long y)
{
	long long ny=ksm(y,P-2),sum=1,s=ksm(G,m);
	for(int i=0;i<m;i++)
	{
		long long val=(sum=sum*s%P)*ny%P;
        arr tmp;
        tmp.x=val,tmp.id=-1;
        long long pos=lower_bound(E,E+m+1,tmp)-E;
		if(E[pos].x==val) return i*m+m-E[pos].id;
	}
	return 0;
}
int exgcd(int a,int b,int &x,int &y)
{
	if(!b)
	{
		x=1,y=0;
		return a;
	}
	long long r=exgcd(b,a%b,y,x);
	y-=x*(a/b);
	return r;
}
void residue(long long a,long long b,long long m)
{
	int x=0,y=0,d=exgcd(a,m,x,y);
	if(b%d) return;
	num=d;
	sol[0]=x*(b/d)%m;
	for(int i=1;i<d;i++) sol[i]=(sol[i-1]+m/d)%m;
	for(int i=0;i<d;i++) sol[i]=(sol[i]+m)%m;
}
void solve()
{
	pre();
	num=0,residue(A,BSGS(B),P-1);
	if(!num) printf("No Solution\n");
	else
	{
		for(int i=0;i<num;i++) ans[i]=ksm(G,sol[i]);
		sort(ans,ans+num);
		for(int i=0;i<num;i++) printf("%lld ",ans[i]);
		putchar('\n');
	}
}
int main()
{
	for(scanf("%d",&T);T--;)
	{
		scanf("%lld%lld%lld",&P,&A,&B);
		solve();
	}
}