結果

問題 No.2374 ASKT Subsequences
ユーザー NP
提出日時 2023-07-08 08:10:52
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,176 bytes
コンパイル時間 1,782 ms
コンパイル使用メモリ 129,704 KB
実行使用メモリ 13,884 KB
最終ジャッジ日時 2024-07-22 04:04:09
合計ジャッジ時間 6,210 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23 WA * 1 TLE * 1 -- * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <unordered_map>
#include <vector>

int count_asakatsu_subsequences(int N, std::vector<int>& A) {
    int cnt = 0;
    std::unordered_map<int, std::vector<int>> indices;

    for (int i = 0; i < N; i++) {
        int a = A[i];
        indices[a].push_back(i);
    }

    for (int i = 0; i < N - 3; i++) {
        for (int j = i + 1; j < N - 2; j++) {
            int k = A[j] - A[i] - 10;
            if (k > 0) {
                std::vector<int>& m_values = indices[A[j] - k];
                for (int m : m_values) {
                    if (m > j) {
                        std::vector<int>& n_values = indices[A[m] + k + 1];
                        for (int n : n_values) {
                            if (n > m) {
                                cnt++;
                            }
                        }
                    }
                }
            }
        }
    }

    return cnt;
}

int main() {
    int N;
    std::cin >> N;
    std::vector<int> A(N);
    for (int i = 0; i < N; i++) {
        std::cin >> A[i];
    }

    int result = count_asakatsu_subsequences(N, A);

    std::cout << result << std::endl;

    return 0;
}
0