Submission #752930

#	Time	Username	Problem	Language	Result	Execution time	Memory
752930		I_love_Hoang_Yen	Sequence (APIO23_sequence)	C++17	0 / 100	2075 ms	10060 KiB

sequence

#include "sequence.h"
#include <bits/stdc++.h>
#define SZ(s) ((int) ((s).size()))
using namespace std;

int sequence(int n, std::vector<int> a) {
    int res = 0;
    int ln = *max_element(a.begin(), a.end());
    for (int median = 0; median <= ln; ++median) {
        vector<int> cnt_less(n, 0);
        vector<int> cnt_equal(n, 0);
        vector<int> cnt_greater(n, 0);
        for (int i = 0; i < n; ++i) {
            cnt_less[i] = a[i] < median;
            cnt_equal[i] = a[i] == median;
            cnt_greater[i] = a[i] > median;
        }
        std::partial_sum(cnt_less.begin(), cnt_less.end(), cnt_less.begin());
        std::partial_sum(cnt_equal.begin(), cnt_equal.end(), cnt_equal.begin());
        std::partial_sum(cnt_greater.begin(), cnt_greater.end(), cnt_greater.begin());

        auto f = [&] (int i) {
            // +n to make sure result >= 0
            if (i < 0) return 0+n;
            return cnt_less[i] + cnt_equal[i] - cnt_greater[i] + n;
        };

        for (int l = 0; l < n; ++l) {
            for (int r = l + 1; r < n; ++r) {
                if (f(r) >= f(l-1)) {
                    int cnt = cnt_equal[r] - (l > 0 ? cnt_equal[l-1] : 0);
                    res = max(res, cnt);
                }
            }
        }
    }
    return res;
}

• Subtask 1: 0 / 11
• Subtask 2: 0 / 17
• Subtask 3: 0 / 7
• Subtask 4: 0 / 12
• Subtask 5: 0 / 13
• Subtask 6: 0 / 22
• Subtask 7: 0 / 18