結果

問題 No.2709 1975 Powers
コンテスト
ユーザー GOTKAKO
提出日時 2024-03-31 15:20:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 754 ms / 2,000 ms
コード長 957 bytes
コンパイル時間 2,199 ms
コンパイル使用メモリ 203,704 KB
最終ジャッジ日時 2025-02-20 17:51:17
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

long long repeat2(long long a, long long b,long long mod){
    long long p = a%mod,ret = 1;
    while(b){
        if(b&1) ret = ret*p %mod;
        p = p*p %mod; b >>= 1;
    }
    return ret;
}

int main() {
    int N,P,Q; cin >> N >> P >> Q;
    vector<int> A(N);
    for(auto &a : A) cin >> a;
    sort(A.begin(),A.end());

    vector<vector<int>> Rmod(N);
    vector<int> add(100000);
    for(int i=N-1; i>=0; i--) add.at(repeat2(5,A.at(i),P))++,Rmod.at(i) = add;
    
    long long answer = 0;
    for(int i=0; i<N-3; i++){
        long long a = repeat2(10,A.at(i),P);
        for(int k=i+1; k<N-2; k++){
            long long b = repeat2(9,A.at(k),P);
            for(int l=k+1; l<N-1; l++){
                long long c = repeat2(7,A.at(l),P);
                long long goal = (Q-(a+b+c)%P+P)%P;
                answer += Rmod.at(l+1).at(goal);
            }
        }
    }
    cout << answer << endl;
}
0