結果

問題 No.1140 EXPotentiaLLL!
ユーザー kurotemko
提出日時 2020-08-03 20:05:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,515 bytes
コンパイル時間 2,136 ms
コンパイル使用メモリ 197,044 KB
最終ジャッジ日時 2025-01-12 13:50:13
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 4 WA * 8
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:55:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   55 |         scanf("%lld %d", &a, &p);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

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

long long mod = 1000000007;
vector<bool> vp;
vector<int> vk(5000001);

// ------------------------------
// 素数列挙
// 参考：https://ei1333.github.io/luzhiled/snippets/math/prime-table.html
void prime_table(int n) {
  vector<bool> prime(n+1, true);
  if(n >= 0) prime[0] = false;
  if(n >= 1) prime[1] = false;
  for(int i = 2; i*i <= n; ++i) {
    if(!prime[i]) continue;
    for(int j = i+i; j <= n; j += i) {
      prime[j] = false;
    }
  }
  vp = prime;
}
// ------------------------------

void kaijou() {
    vk[0] = 1;
    for(int i = 1; i <= 5000000; ++i) vk[i] = (vk[i-1]*i)%mod;
}

// ------------------------------
// 繰り返し二乗法
// 参考：https://math.nakaken88.com/textbook/cp-binary-exponentiation-and-recursive-function/
int modPow(int a, int n, int p) {
  if (n == 0) return 1;
  if (n == 1) return a % p;
  if (n % 2 == 1) return (a * modPow(a, n - 1, p)) % p;
  long long t = modPow(a, n / 2, p);
  return (t * t) % p;
}
// ------------------------------

int main() {

    // あらかじめ素数テーブルと階乗テーブルを作成
    // ちゃんとmodを取っておく
    prime_table(5000000);
    kaijou();

    long long t;
    cin >> t;

    for(int i = 0; i < t; ++i) {
        long long a;
        int p;
        scanf("%lld %d", &a, &p);
        if(vp[p]) {
            int ans = modPow((a%p), vk[p], p);
            printf("%d\n", ans);
        }
        else {
            printf("-1\n");
        }
    }
}
0