[SDOI2011]计算器 BSGS

前两个操作都看出来做法了，第三个要用到BSGS这个算法

BSGS主要可以解 a^x=b(mod n)的 0<=x<n 的解

暂时先拿kb和hzwer的板子当黑盒吧。。数论细节短时间内没有学的打算

upd:

学了一下BSGS算法的的理论以及这个板子的实现，似乎这个板子默认模数素数

模数不素的话用kb的吧，那个似乎是用前向星处理哈希冲突做的。。

#include<bits/stdc++.h>#include<stdio.h>#include<algorithm>#include<queue>#include<string.h>#include<iostream>#include<math.h>#include<set>#include<map>#include<vector>#include<iomanip>using namespace std;#define ll long long#define ull unsigned long long#define pb push_back#define FOR(a) for(int i=1;i<=a;i++)const int inf=0x3f3f3f3f;const ll Linf=9e18;const int maxn=1e5+7; const ll mod=100003;const double eps=1e-6;int gcd(int a,int b){return b==0?a:gcd(b,a%b);}void exgcd(int a,int b,int &x,int &y){if(b==0){x=1;y=0;return;}exgcd(b,a%b,x,y);int t=x;x=y;y=t-a/b*y;}int work1(ll y,int z,int p){y%=p;ll ans=1;for(int i=z;i;i>>=1,y=y*y%p)if(i&1)ans=ans*y%p;//printf("%d\n",ans);return ans;}void work2(int y,int z,int p){//xy+tp=z,求x最小正数解int g=gcd(y,p);if(z%g){puts("Orz, I cannot find x!");return;}y/=g;z/=g;p/=g;int a,b;exgcd(y,p,a,b);a=(ll)a*z%p;while(a<0)a+=p;printf("%d\n",a);}map<int,int>mp;void work3(int y,int z,int p){y%=p;if(!y&&!z){puts("1");return;}if(!y){puts("Orz, I cannot find x!");return;}mp.clear();ll m=ceil(sqrt(p)),t=1;mp[1]=m+1;for(ll i=1;i<m;i++){t=t*y%p;if(!mp[t])mp[t]=i;}ll tmp=work1(y,p-m-1,p),ine=1;for(ll k=0;k<m;k++){int i=mp[z*ine%p];if(i){if(i==m+1)i=0;printf("%lld\n",k*m+i);return;}ine=ine*tmp%p;}puts("Orz, I cannot find x!");}int main(){int T,K;scanf("%d%d",&T,&K);for(int i=1;i<=T;i++){int y,z,p;scanf("%d%d%d",&y,&z,&p);if(K==1){int ans=work1(y,z,p);printf("%d\n",ans);}else if(K==2){work2(y,z,p);}else{work3(y,z,p);}}}