Baby Step,Giant Step算法模板

介绍：BSGS（Baby Step，Giant Step）算法是用来求高次同余方程的解的算法。

模板：P148

//baby step giant step
#include<iostream>
#include<map>
#include<cmath>
using namespace std;
typedef long long ll;
ll q_pow(ll a,ll b,ll M){
	ll res = 1;
	while(b > 0){
		if(b&1) res = res*a%M;
		a = a*a%M;
		b >>= 1;
	}
	return res%M;
}
int BSGS(ll a,ll b,ll p){
	map<int,int> hash;
	hash.clear();
	int t = (int)sqrt(p)+1;
	for(int j = 0;j < t;j++){
		int val = b*q_pow(a,j,p)%p;
		hash[val] = j;
	}	
	a = q_pow(a,t,p);//a = a^t
	for(int i = 0;i <= t;i++){
		int val = q_pow(a,i,p);
		if(hash.count(val) && i*t-hash[val]>0)
			return i*t-hash[val];//hash[val] = j;
	}
	return -1;//无解 
}
int main(){
	int a,b,x,p;
	cin >> a >> b >> p;
	cout << BSGS(a,b,p);
	return 0;
}