【BZOJ】3239: Discrete Logging

http://www.lydsy.com/JudgeOnline/problem.php?id=3239

题意：原题很清楚了= =

#include <bits/stdc++.h>
using namespace std;

map<int, int> s;
typedef long long ll;
int mpow(int a, int b, int p) {
	a%=p; int r=1;
	while(b) { if(b&1) r=((ll)r*a)%p; a=((ll)a*a)%p; b>>=1; }
	return r;
}
void work(int a, int b, int p) {
	a%=p; b%=p;
	if(b==1) { puts("0"); return; }
	if(!a && !b) { puts("1"); return; }
	if(!a) { puts("no solution"); return; }
	s.clear();
	int m=sqrt(p+0.5), t=1, w=a, mm;
	for(int i=0; i<m; ++i) s[((ll)b*t)%p]=i, t=((ll)t*w)%p;
	w=mpow(a, m, p); t=1; mm=(p-1)/m+1; bool flag=1;
	for(int i=0; i<=mm; ++i) if(s.count(t) && ((ll)m*i-s[t])>=0) { printf("%lld\n", (ll)m*i-s[t]); flag=0; break; } else t=((ll)t*w)%p;
	if(flag) puts("no solution");
}
int main() {
	int a, b, p;
	while(~scanf("%d%d%d", &p, &a, &b)) work(a, b, p);
	return 0;
}

bsgs裸题= =