結果

問題 No.551 夏休みの思い出（２）
ユーザー miyo2580miyo2580
提出日時 2024-02-21 00:25:59
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,935 bytes
コンパイル時間 2,452 ms
コンパイル使用メモリ 213,904 KB
実行使用メモリ 35,492 KB
最終ジャッジ日時 2024-09-29 03:56:36
合計ジャッジ時間 15,232 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 21 ms
13,644 KB
testcase_01 AC 26 ms
6,820 KB
testcase_02 WA -
testcase_03 AC 85 ms
6,816 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 23 ms
6,820 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 TLE -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
testcase_45 -- -
testcase_46 -- -
testcase_47 -- -
testcase_48 -- -
権限があれば一括ダウンロードができます

ソースコード

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;
    map<ll,ll> mp;
    mint memo=1;
    rep(i,1000)memo*=r;
    mint now=1;
    rep(i,1000000){
        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,1001){
            x*=r;
            if(mp.count(x.x)){
                ll X=mp[now.x]*1000;
                X-=i;
                if(X%2!=0)continue;
                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