結果
問題 | No.2709 1975 Powers |
ユーザー |
![]() |
提出日時 | 2024-03-31 13:56:59 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,067 ms / 2,000 ms |
コード長 | 1,149 bytes |
コンパイル時間 | 2,576 ms |
コンパイル使用メモリ | 212,568 KB |
最終ジャッジ日時 | 2025-02-20 16:45:28 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 25 |
ソースコード
#include <bits/stdc++.h> using namespace std; long long f(long long x, long long n, long long MOD) { long long ret = 1; while(n) { if(n&1) { ret *= x; ret %= MOD; } x *= x; x %= MOD; n >>= 1; } return ret; } int main() { long long N, P, Q; cin >> N >> P >> Q; vector<long long> A(N); for(int i = 0; i < N; i++) { cin >> A[i]; } sort(A.begin(), A.end()); vector cnt(N+1, unordered_map<long long, long long>()); for(int i = N-1; i >= 0; i--) { cnt[i] = cnt[i+1]; cnt[i][f(5, A[i], P)]++; } vector po(N, vector<long long>(15)); for(int i = 0; i < N; i++) { for(int j = 0; j < 15; j++) { po[i][j] = f(j, A[i], P); } } long long ans = 0; for(int i = 0; i < N; ++i) { for(int j = i+1; j < N; ++j) { for(int k = j+1; k < N; ++k) { long long s = po[i][10]+po[j][9]+po[k][7]; s %= P; long long t = (P+Q-s)%P; // cout << A[i] << " " << A[j] << " " << A[k] << ": " << f(10, A[i], P) << " " << f(9, A[j], P) << " " << f(7, A[k], P) << " " << t << endl; ans += cnt[k+1][t]; } } } cout << ans << endl; }