洛谷P2485 [SDOI2011]计算器

case 1 :快速幂取模 ；
case 2:exgcd求乘法逆元，注意y%p的逆元存在条件为gcd(y,p) = 1 ，即y 与 p 互质;
case 3:扩展BSGS （因为题中没说y 与 p 互质）。

源码：

#include <iostream>
#include <algorithm>
#include <map>
#include <cmath>
using namespace std;

long long t,k;
long long y,z,p;

long long powl(long long x,long long n,long long mod){
	if(n == 0) return 1;
	if(n == 1) return x%p;
	if(n&1) return powl(x*x%mod,n/2,mod)*x%mod;
	else return powl(x*x%mod,n/2,mod)%mod;
}


long long gcd(long long a,long long b){
	return b == 0 ? a:gcd(b,a%b);
}

void exgcd(long long a,long long b,long long & xx ,long long & yy){
	if(b == 0){
		xx = 1;
		yy = 0;
		return ;
	}
	exgcd(b,a%b,yy,xx);
	yy -= a/b*xx;
}

void solve_1(){
	while(t--){
		cin >> y >> z >> p;
		cout << powl(y,z,p)%p << endl;;
	}
}

void solve_2(){
	while(t--){
		cin >> y >> z >> p;
		long long xx,yy;
		exgcd(y,p,xx,yy);
		if(gcd(y,p) == 1) cout << z*(xx%p+p)%p << endl;
		else cout << "Orz, I cannot find x!" << endl;
	}
}

void solve_3(){
	while(t--){
		map<long long,long long> mp;
		cin >> y >> z >> p;
		long long m = sqrt(p);
		long long x = z%p;
		y %= p;
		long long u = 1;
		for(long long i = 0;i<m;i++){
			mp[x] = i;
			x = (x*y)%p;
			u = (u*y)%p;
		}
		long long ans = 1;
		long long flag = 0;
		for(long long i = 0;i <= m+1;i++){
			if(mp.find(ans) != mp.end() ) {
				cout << i*m-mp[ans] << endl;
				flag = 1;
				break;
			}
			if(u == 0) break;
			ans = (ans*u)%p;
		}
		if(!flag) cout << "Orz, I cannot find x!" << endl;
	}
}

int main(){
	cin >> t >> k;
	switch(k){
		case 1:solve_1();break;
		case 2:solve_2();break;
		case 3:solve_3();break;
	}
	return 0;
}

此处case未用扩展BSGS ， 只是在普通BSGS基础上加了一条语句：

if(u == 0) break;

即巧妙地解决了问题。

附加：扩展BSGS公式：

y ^ k /d * y ^(x-k) ≡ n/d (mod p/d) 。然后化为一般的BSGS， 求出结果后还原。 即 x = a*m-b+k 。