未知原因,导致一样的代码出现了不同的评测结果
查看原帖
未知原因,导致一样的代码出现了不同的评测结果
656765
STARSczy楼主2024/10/29 15:46

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;
}
2024/10/29 15:46
加载中...