Submission #998016

#	Time	Username	Problem	Language	Result	Execution time	Memory
998016		Qwerty1232	Cigle (COI21_cigle)	C++17	100 / 100	310 ms	166080 KiB

cigle

#include <bits/stdc++.h>

int32_t main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int n;
    std::cin >> n;
    std::vector<int> input(n);
    for (auto& i : input) {
        std::cin >> i;
    }
    std::vector<int> prf(n + 1);
    for (int i = 0; i < n; i++) {
        prf[i + 1] = prf[i] + input[i];
    }

    std::vector<std::vector<int>> dp(n, std::vector<int>(n));
    std::vector<std::array<int, 4>> vec;
    for (int i = 1; i < n; i++) {
        int dlt = 0;
        int sum = 0;
        for (int j = i, j2 = i; j < n; j++) {
            sum += input[j];
            while (j2 > 0 && prf[i] - prf[j2] < sum) {
                j2--;
            }
            if (j2 > 0 && j + 1 < n && prf[i] - prf[j2] == sum) {
                dlt++;
                vec.push_back({j2 - 1, i, j + 1, dlt});
            }
        }
    }
    std::sort(vec.begin(), vec.end(), [&](auto a, auto b) { return a[1] < b[1]; });
    int vec_ptr = 0;
    for (int j = 0; j < n; j++) {
        for (int i = 0; i <= j; i++) {
            if (i > 0) {
                dp[i][j] = std::max(dp[i][j], dp[i - 1][j]);
            }
            if (i < j) {
                dp[i][j] = std::max(dp[i][j], dp[i][j - 1]);
            }
        }

        while (vec_ptr < vec.size() && vec[vec_ptr][1] == j) {
            auto [a, b, c, dlt] = vec[vec_ptr++];
            // for (int t = 0; c + t < n; t++) {
            for (int t = 0; t < 1; t++) {
                dp[b][c + t] = std::max(dp[b][c + t], dp[a][b - 1] + dlt);
            }
        }
    }

    int ans = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            ans = std::max(ans, dp[i][j]);
        }
    }

    std::cout << ans << "\n";

    return 0;
}

Compilation message (stderr)

cigle.cpp: In function 'int32_t main()':
cigle.cpp:46:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::array<int, 4> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |         while (vec_ptr < vec.size() && vec[vec_ptr][1] == j) {
      |                ~~~~~~~~^~~~~~~~~~~~

• Subtask 1: 9 / 9
• Subtask 2: 11 / 11
• Subtask 3: 13 / 13
• Subtask 4: 15 / 15
• Subtask 5: 52 / 52