Submission #72206646

Source Code Expand

#include <iostream>
#include <vector>
#include <map>

using namespace std;
using ll = long long;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int N;
    cin >> N;
    vector<ll> A(N);
    map<ll, int> total_cnt;
    for (int i = 0; i < N; i++) {
        cin >> A[i];
        total_cnt[A[i]]++;
    }
    map<ll, int> left_cnt;
    ll ans = 0;
    for (int j = 0; j < N; j++) {
        if (A[j] % 5 != 0) {
            left_cnt[A[j]]++;
            continue;
        }
        ll m = A[j] / 5;
        ll val7 = 7 * m;
        ll val3 = 3 * m;
        ll left7 = left_cnt[val7];
        ll left3 = left_cnt[val3];
        ll right7 = total_cnt[val7] - left7 - (A[j] == val7 ? 1 : 0);
        ll right3 = total_cnt[val3] - left3 - (A[j] == val3 ? 1 : 0);
        ans += right7 * right3;
        ans += left7 * left3;
        left_cnt[A[j]]++;
    }
    cout << ans;
    return 0;
}

Submission Info

Submission Time
Task D - Kadomatsu Subsequence
User WoodyLee
Language C++23 (GCC 15.2.0)
Score 425
Code Size 969 Byte
Status AC
Exec Time 659 ms
Memory 58300 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 3628 KiB
sample_02.txt AC 1 ms 3560 KiB
sample_03.txt AC 1 ms 3528 KiB
test_01.txt AC 1 ms 3544 KiB
test_02.txt AC 1 ms 3448 KiB
test_03.txt AC 1 ms 3408 KiB
test_04.txt AC 1 ms 3528 KiB
test_05.txt AC 18 ms 5704 KiB
test_06.txt AC 18 ms 5580 KiB
test_07.txt AC 18 ms 5616 KiB
test_08.txt AC 18 ms 5692 KiB
test_09.txt AC 18 ms 5748 KiB
test_10.txt AC 19 ms 5684 KiB
test_11.txt AC 21 ms 5704 KiB
test_12.txt AC 22 ms 5832 KiB
test_13.txt AC 18 ms 5704 KiB
test_14.txt AC 26 ms 5752 KiB
test_15.txt AC 26 ms 5708 KiB
test_16.txt AC 24 ms 5692 KiB
test_17.txt AC 32 ms 5756 KiB
test_18.txt AC 40 ms 5684 KiB
test_19.txt AC 37 ms 5656 KiB
test_20.txt AC 175 ms 8336 KiB
test_21.txt AC 91 ms 6204 KiB
test_22.txt AC 151 ms 8012 KiB
test_23.txt AC 659 ms 30324 KiB
test_24.txt AC 617 ms 30280 KiB
test_25.txt AC 288 ms 19532 KiB
test_26.txt AC 515 ms 58056 KiB
test_27.txt AC 504 ms 58276 KiB
test_28.txt AC 497 ms 58148 KiB
test_29.txt AC 501 ms 58300 KiB
test_30.txt AC 505 ms 57988 KiB