結果
問題 | No.551 夏休みの思い出(2) |
ユーザー |
![]() |
提出日時 | 2019-04-13 17:52:53 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2,240 ms / 4,000 ms |
コード長 | 2,327 bytes |
コンパイル時間 | 2,388 ms |
コンパイル使用メモリ | 200,256 KB |
最終ジャッジ日時 | 2025-01-07 02:09:24 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 47 |
ソースコード
#include<bits/stdc++.h> using namespace std; using Int = long long; template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;} template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;} Int MOD = 1000000007; template<typename T> struct Mint{ T v; Mint():v(0){} Mint(signed v):v(v){} Mint(long long t){v=t%MOD;if(v<0) v+=MOD;} Mint pow(long long k){ Mint res(1),tmp(v); while(k){ if(k&1) res*=tmp; tmp*=tmp; k>>=1; } return res; } Mint inv(){return pow(MOD-2);} Mint& operator+=(Mint a){v+=a.v;if(v>=MOD)v-=MOD;return *this;} Mint& operator-=(Mint a){v+=MOD-a.v;if(v>=MOD)v-=MOD;return *this;} Mint& operator*=(Mint a){v=1LL*v*a.v%MOD;return *this;} Mint& operator/=(Mint a){return (*this)*=a.inv();} Mint operator+(Mint a) const{return Mint(v)+=a;}; Mint operator-(Mint a) const{return Mint(v)-=a;}; Mint operator*(Mint a) const{return Mint(v)*=a;}; Mint operator/(Mint a) const{return Mint(v)/=a;}; Mint operator-(){return v?MOD-v:v;} bool operator==(const Mint a)const{return v==a.v;} bool operator!=(const Mint a)const{return v!=a.v;} bool operator <(const Mint a)const{return v <a.v;} // find x s.t. a^x = b static T log(Mint a,Mint b){ static T sq = 1000000; static unordered_map<T, T> dp; dp.reserve(sq); if(dp.empty()){ Mint res(1); for(int r=0;r<sq;r++){ if(!dp.count(res.v)) dp[res.v]=r; res*=a; } } static Mint p=a.inv().pow(sq); Mint res=b; for(int q=0;q<=MOD/sq+1;q++){ if(dp.count(res.v)){ T idx=q*sq+dp[res.v]; if(idx>0) return idx; } res*=p; } return T(-1); } }; //INSERT ABOVE HERE signed main(){ cin.tie(0); ios::sync_with_stdio(0); using M = Mint<int>; int r; cin>>MOD>>r; M R(r); int q; cin>>q; M TI=M(2).inv(); for(int i=0;i<q;i++){ int a,b,c; cin>>a>>b>>c; M A(a),B(b),C(c); M AI=A.inv(); M X=AI*B*TI; M D=X*X-AI*C; M E=-AI*B*TI; if(D.v==0){ cout<<E.v<<"\n"; continue; } int k=M::log(R,D); if(k&1){ cout<<-1<<"\n"; continue; } M F=R.pow(k/2); int x=(E+F).v,y=(E-F).v; if(x>y) swap(x,y); cout<<x<<" "<<y<<"\n"; } cout<<flush; return 0; }