結果

問題 No.551 夏休みの思い出（２）
コンテスト
ユーザー 259_Momone
提出日時 2017-07-29 00:02:04
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 48 ms / 4,000 ms
コード長 1,949 bytes
コンパイル時間 1,540 ms
コンパイル使用メモリ 159,740 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-10 06:56:51
合計ジャッジ時間 3,948 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 47
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:40:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   40 |     scanf("%lld%lld%lld", &P, &R, &Q);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:43:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   43 |         scanf("%lld%lld%lld", &A, &B, &C);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

long long int P, R, Q, A, B, C, D, x, y, it;

static inline long long mdp(long long int a, long long int n){
    long long int r = 1;
    while(n){
        if(n & 1)r = r * a % P;
        a = a * a % P;
        n /= 2;
    }
    return r;
}

static inline long long pmdp(long long a, long long b){
    long long int r = 1, s = 0, i1, i2, ro = a, n = (P + 1) / 2;
    swap(a, b);
    b = 1;
    //cout << " | " << a << " " << b << " (" << a << " + sqrt(" << ro << ")) ";
    while(n){
        //cout << "(" << n << ", " << r << ", " << s << ", " << a << ", " << b << ") ";
        if(n & 1){
            i1 = (a * r + (b * s % P) * ro) % P;
            i2 = (a * s + b * r) % P;
            r = i1;
            s = i2;
        }
        i1 = (a * a + (b * b % P) * ro) % P;
        i2 = (2 * a * b) % P;
        a = i1;
        b = i2;
        n /= 2;
    }
    //cout << r << " " << s << " | ";
    return r;
}

int main(){
    scanf("%lld%lld%lld", &P, &R, &Q);
    it = mdp(2, P - 2);
    for(int Qi = 0; Qi < Q; ++Qi){
        scanf("%lld%lld%lld", &A, &B, &C);
        B = B * mdp(A, P - 2) % P;
        C = C * mdp(A, P - 2) % P;
        A = 1;
        //cout << "X^2 + " << B << " X + " << C << " = 0 (mod " << P << ") ";
        D = (B * B + 4 * (P - 1) * C) % P;
        //cout << D << " ";
        if(mdp(D, (P - 1) / 2) == P - 1){puts("-1");continue;}
        if(D == 0){
            x = 0;
            y = (((x + (P - 1) * B) % P) * it) % P;
            cout << y << endl;
        }else{
            int i;
            for(i = 0; i < P; ++i)if(mdp((i * i + (P - 1) * D) % P, (P - 1) / 2) == P - 1){y = (i * i + (P - 1) * D) % P;break;}
            x = pmdp(y, i);
            //cout << x << " ";
            y = (((x + (P - 1) * B) % P) * it) % P;
            x = (((P - 1) * (x + B) % P) * it) % P;
            cout << min(x, y) << " " << max(x, y) << endl;
        }
    }
    return 0;
}
0