結果

問題 No.551 夏休みの思い出（２）
ユーザー miyo2580
提出日時 2024-02-21 01:00:55
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 3,007 bytes
コンパイル時間 2,536 ms
コンパイル使用メモリ 210,596 KB
最終ジャッジ日時 2025-02-19 18:26:47
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other AC * 6 TLE * 1 -- * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define repd(i,a,b) for (ll i=(a);i<(b);i++)
#define rep(i,n) repd(i,0,n)
#define all(x) (x).begin(),(x).end()
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; }
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; }
typedef long long ll;
typedef pair<ll,ll> P;
typedef vector<ll> vec;
using Graph = vector<vector<ll>>;
const long long INF = 1LL<<60;
const long long MOD = 1000000007;

// auto mod int
// https://youtu.be/L8grWxBlIZ4?t=9858
// https://youtu.be/ERZuLAxZffQ?t=4807 : optimize
// https://youtu.be/8uowVvQ_-Mo?t=1329 : division
int mod = 1000000007;
struct mint {
  ll x; // typedef long long ll;
  mint(ll x=0):x((x%mod+mod)%mod){}
  mint operator-() const { return mint(-x);}
  mint& operator+=(const mint a) {
    if ((x += a.x) >= mod) x -= mod;
    return *this;
  }
  mint& operator-=(const mint a) {
    if ((x += mod-a.x) >= mod) x -= mod;
    return *this;
  }
  mint& operator*=(const mint a) { (x *= a.x) %= mod; return *this;}
  mint operator+(const mint a) const { return mint(*this) += a;}
  mint operator-(const mint a) const { return mint(*this) -= a;}
  mint operator*(const mint a) const { return mint(*this) *= a;}
  mint pow(ll t) const {
    if (!t) return 1;
    mint a = pow(t>>1);
    a *= a;
    if (t&1) a *= *this;
    return a;
  }

  // for prime mod
  mint inv() const { return pow(mod-2);}
  mint& operator/=(const mint a) { return *this *= a.inv();}
  mint operator/(const mint a) const { return mint(*this) /= a;}
};
istream& operator>>(istream& is, mint& a) { return is >> a.x;}
ostream& operator<<(ostream& os, const mint& a) { return os << a.x;}

int main()
{  
    ios::sync_with_stdio(false);
    cin.tie(0);
    ll p,r;
    cin>>mod>>r;
    ll q;cin>>q;
    unordered_map<ll,ll> mp;
    mint memo=1;
    rep(i,10000)memo*=r;
    mint now=1;
    rep(i,100000){
        now*=memo;
        if(mp.count(now.x))continue;
        mp[now.x]=i+1;
    }
    rep(qi,q){
        ll A,B,C;
        mint a,b,c;
        cin>>A>>B>>C;
        a=A;
        b=B;
        c=C;
        b/=a;
        c/=a;
        mint x=b;
        x*=b;
        x/=4;
        x-=c;
        if(x.x==0){
            mint ans=0;
            ans-=b/2;
            cout<<ans<<'\n';
            continue;
        }
        vec ans;
        repd(i,1,10000){
            x*=r;
            if(mp.count(x.x)){
                ll X=mp[x.x]*10000;
                X-=i;
                if(X%2!=0)continue;
                ans.push_back(X/2);
                X+=mod-1;
                ans.push_back(X/2);
            }
        }
        map<ll,ll> xs;
        rep(i,ans.size()){
            mint g=mint(r).pow(ans[i]);
            g-=b/2;
            xs[g.x]=1;
        }
        if(xs.size()==0){
            cout<<-1<<'\n';
            continue;
        }
        for(auto[X,_]:xs){
            cout<<X<<' ';
        }
        cout<<'\n';
    }
    return 0;
}
0