Submission #49547718

---

Source Code Expand

/**
 *  @the_hyp0cr1t3
 *  21.01.2024 17:58
**/
#include <bits/stdc++.h>

using i64 = long long;

constexpr int N = 30;

int main() {
    std::cin.tie(nullptr)->sync_with_stdio(false);

    std::vector<std::array<int, 3>> triples;
    for (int i = 1; i <= 10; i++)
        for (int j = 1; j <= 10; j++)
            for (int k = 1; k <= 10; k++)
                if (j - i == k - j) triples.push_back({i, j, k});

    int n;
    std::cin >> n;
    std::vector<int> a(n);
    for (auto &x : a)
        std::cin >> x;

    i64 ans = 1LL * n * (n + 1) / 2;
    for (int i = 0; i < n; i++) {
        std::vector<int> p(triples.size());
        for (int j = i; j < n and j - i < N; j++) {
            for (int t = 0; t < triples.size(); t++)
                p[t] += (a[j] == triples[t][p[t]]);
            if (std::ranges::max(p) == 3) break;
            ans--;
        }
    }

    std::cout << ans << '\n';
}

Submission Info

Compile Error

Main.cpp: In function ‘int main()’:
Main.cpp:30:31: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<std::array<int, 3> >::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   30 |             for (int t = 0; t < triples.size(); t++)
      |                             ~~^~~~~~~~~~~~~~~~

Judge Result