Submission #72215344

Source Code Expand

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

int main() {
    int N;
    cin >> N;

    vector<long long> A(N);
    unordered_map<long long, vector<int>> pos;

    for (int i = 0; i < N; i++) {
        cin >> A[i];
        pos[A[i]].push_back(i);
    }
    long long ans = 0;
    for (int j = 0; j < N; j++) {
        if (A[j] % 5 != 0) continue;
        long long t = A[j] / 5;
        long long v7 = 7 * t;
        long long v3 = 3 * t;
        if (!pos.count(v7) || !pos.count(v3)) continue;
        auto &P7 = pos[v7];
        auto &P3 = pos[v3];
        long long l7 = lower_bound(P7.begin(), P7.end(), j) - P7.begin();
        long long l3 = lower_bound(P3.begin(), P3.end(), j) - P3.begin();
        long long r7 = P7.size() - l7;
        long long r3 = P3.size() - l3;
        ans += l7 * l3 + r7 * r3;
    }
    cout << ans << "\n";
    return 0;
}

Submission Info

Submission Time
Task D - Kadomatsu Subsequence
User akito_
Language C++23 (GCC 15.2.0)
Score 425
Code Size 892 Byte
Status AC
Exec Time 250 ms
Memory 31860 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 425 / 425
Status
AC × 3
AC × 33
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt
All sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt, test_19.txt, test_20.txt, test_21.txt, test_22.txt, test_23.txt, test_24.txt, test_25.txt, test_26.txt, test_27.txt, test_28.txt, test_29.txt, test_30.txt
Case Name Status Exec Time Memory
sample_01.txt AC 1 ms 3504 KiB
sample_02.txt AC 1 ms 3692 KiB
sample_03.txt AC 1 ms 3600 KiB
test_01.txt AC 1 ms 3596 KiB
test_02.txt AC 1 ms 3432 KiB
test_03.txt AC 1 ms 3600 KiB
test_04.txt AC 1 ms 3504 KiB
test_05.txt AC 89 ms 7364 KiB
test_06.txt AC 83 ms 7280 KiB
test_07.txt AC 81 ms 7376 KiB
test_08.txt AC 86 ms 7268 KiB
test_09.txt AC 88 ms 7220 KiB
test_10.txt AC 84 ms 7296 KiB
test_11.txt AC 88 ms 7448 KiB
test_12.txt AC 96 ms 7260 KiB
test_13.txt AC 87 ms 7004 KiB
test_14.txt AC 93 ms 7276 KiB
test_15.txt AC 97 ms 7164 KiB
test_16.txt AC 94 ms 7428 KiB
test_17.txt AC 84 ms 7332 KiB
test_18.txt AC 98 ms 7480 KiB
test_19.txt AC 91 ms 7332 KiB
test_20.txt AC 112 ms 8692 KiB
test_21.txt AC 96 ms 7644 KiB
test_22.txt AC 106 ms 8684 KiB
test_23.txt AC 174 ms 15076 KiB
test_24.txt AC 195 ms 16072 KiB
test_25.txt AC 175 ms 15040 KiB
test_26.txt AC 236 ms 31852 KiB
test_27.txt AC 230 ms 31856 KiB
test_28.txt AC 250 ms 31816 KiB
test_29.txt AC 246 ms 31860 KiB
test_30.txt AC 232 ms 31840 KiB