rt,c++ 17 及以上能过,14 及一下过不了。
把 26 行 unordered_map<int,int> 改成 unordered_map<signed,signed> 好像每个版本都可以过。
求问原因。
#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/rope>
#define rep(i,l,r) for(int i=(l),i##end=(r);i<=i##end;++i)
#define per(i,r,l) for(int i=(r),i##end=(l);i>=i##end;--i)
#define int long long
#define double long double
#define pii pair<int,int>
#define fi first
#define se second
#define pb push_back
#define rbtree(way) tree<way,null_type,less<way>,rb_tree_tag,tree_order_statistics_node_update>
using namespace std;
using namespace __gnu_cxx;
using namespace __gnu_pbds;
const int maxn=1e5+10,maxm=1e6+10,mod=998244353,inf=INT_MAX;
inline int ksm(int x,int k,int mod=mod){
int ans=1;
for(;k;k>>=1,x=x*x%mod) if(k&1) ans=ans*x%mod;
return ans;
}
namespace fbsgs{
int p,g,ql,hp,qlp=1,tmp=1,pt,pr[maxn],fl[maxn];
unordered_map<int,int> t;
inline int maker(int x){
int cnt=0;
for(;!t.count(x);(x*=g)%=p,++cnt);
return (t[x]-cnt+p-1)%(p-1);
}
inline void init(){
hp=sqrt(p),ql=sqrt(sqrt(p)*log(p));
rep(i,1,ql) (qlp*=g)%=p;
rep(i,0,p/ql) t[tmp]=i*ql,(tmp*=qlp)%=p;
rep(i,2,hp+1){
if(!fl[i]) t[i]=fl[i]=maker(i),pr[++pt]=i;
rep(j,1,pt) if(i*pr[j]>iend) j=jend;
else t[i*pr[j]]=fl[i*pr[j]]=(fl[i]+fl[pr[j]])%(p-1),i%pr[j]==0?j=jend:0;
}
t[p-1]=maker(p-1),t[1]=0;
}
inline int query(int x){
if(x<=hp) return t[x];
int v=p/x,r=p%x;
if(r<<1<=x) return (t[p-1]+query(r)+p-1-t[v])%(p-1);
return (query(x-r)+p-1-t[v+1])%(p-1);
}
}
int n;
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>fbsgs::p>>fbsgs::g>>n,fbsgs::init();
rep(i,1,n){
int x=i;
cin>>x,cout<<fbsgs::query(x)<<'\n';
}
return 0;
}