1. 程式人生 > >Baby Step,Giant Step演算法模板

Baby Step,Giant Step演算法模板

介紹:BSGS(Baby Step,Giant Step)演算法是用來求高次同餘方程a^x\equiv b(mod P)的解的演算法。

模板: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;
}