結果

問題 No.2709 1975 Powers
コンテスト
ユーザー 伊藤遼
提出日時 2024-11-01 21:59:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,254 bytes
コンパイル時間 3,390 ms
コンパイル使用メモリ 161,944 KB
実行使用メモリ 9,216 KB
最終ジャッジ日時 2024-11-01 21:59:41
合計ジャッジ時間 11,053 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 1
other AC * 8 WA * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <math.h>
#include <queue>
#include <numeric>
#include <map>
#include <atcoder/all>
using namespace atcoder;
using namespace std;

using ll = long long;

#define Sort(a) sort((a).begin(), (a).end())
#define Rev(a) reverse((a).begin(), (a).end())
#define cfs(a) cout << fixed << setprecision(a)


ll modpow(ll a, ll n, ll mod){
    ll ans = 1;
    while (n > 0){
        if (n & 1) ans = ans * a % mod;
        a = a * a % mod;
        n /= 2;
    }
    return ans;
}

int main()
{
    ll N, P, Q;
    cin >> N >> P >> Q;
    vector<ll> A(N);
    for (ll i = 0; i < N; i++)
    {
        cin >> A[i];
    }
    vector<ll> num_vec = { 10, 9, 7, 5 };
    vector<vector<ll>> modA(num_vec.size(), vector<ll>(N));
    for (ll i = 0; i < num_vec.size(); i++)
    {
        for (ll j = 0; j < N; j++)
        {
            modA[i][j] = modpow(num_vec[i], A[j], P);
        }
    }

    map<ll, ll> mp;
    for (ll i = 0; i < N; i++)
    {
        mp[modA[3][i]]++;
    }
    
    ll ans = 0;
    for (ll i = 0; i < N; i++)
    {
        for (ll j = 0; j < N; j++)
        {
            for (ll k = 0; k < N; k++)
            {
                if (i == j || j == k || k == i)
                {
                    break;
                }
                ll Sum = modA[0][i] + modA[1][j] + modA[2][k];
                Sum %= P;
                ll check_num = 0;
                if (Sum <= Q)
                {
                    check_num += Q - Sum;
                }
                else
                {
                    check_num += (P + Q) - Sum;
                }
                ll temp_cnt = 0;
                if (modA[3][i] == check_num)
                {
                    temp_cnt++;
                }
                if (modA[3][j] == check_num)
                {
                    temp_cnt++;
                }
                if (modA[3][k] == check_num)
                {
                    temp_cnt++;
                }
                // if (mp[check_num] - temp_cnt < 0)
                // {
                //     break;
                // }
                ans += mp[check_num] - temp_cnt;
            }
        }
    }
    cout << ans << endl;
}
0