oj.uz

Submission #989994

# Submission time Handle Problem Language Result Execution time Memory
989994 2024-05-29T10:11:46 Z LOLOLO Kpart (eJOI21_kpart) C++17
100 / 100
 1310 ms 196176 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
 
#define           f     first
#define           s     second
#define           pb    push_back
#define           ep    emplace
#define           eb    emplace_back
#define           lb    lower_bound
#define           ub    upper_bound
#define       all(x)    x.begin(), x.end()
#define      rall(x)    x.rbegin(), x.rend()
#define   uniquev(v)    sort(all(v)), (v).resize(unique(all(v)) - (v).begin())
#define     mem(f,x)    memset(f , x , sizeof(f))
#define        sz(x)    (ll)(x).size()
#define  __lcm(a, b)    (1ll * ((a) / __gcd((a), (b))) * (b))
#define          mxx    *max_element
#define          mnn    *min_element
#define    cntbit(x)    __builtin_popcountll(x)
#define       len(x)    (int)(x.length())
 
const int N = 1e3 + 10;
const int M = 5e4;
int f[N][M + 1], psum[N];

void solve() {
    int n;
    cin >> n;

    int sum = 0;
    for (int i = 1; i <= n; i++) {
        int x;
        cin >> x;
        sum += x;
        psum[i] = sum;
        for (int j = 1; j <= max(sum, M); j++) {
            f[i][j] = f[i - 1][j];
            if (j == x) {
                f[i][j] = i;
            } else if (j > x) {
                f[i][j] = max(f[i][j], f[i - 1][j - x]);
            }
        }
    }

    vector <int> val;
    for (int i = 1; i <= n; i++) {
        int is = 1;
        for (int j = i; j <= n; j++) {
            int sum = psum[j] - psum[j - i];
            if (sum % 2) {
                is = 0;
                break;
            } 

            if (j - f[j][sum / 2] + 1 > i) {
                is = 0;
                break;
            }
        }

        if (is)
            val.pb(i);
    }

    cout << sz(val) << " ";
    for (auto x : val)
        cout << x << " ";

    cout << '\n';
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int t = 1;
    cin >> t;

    while (t--) {
        solve();
    }

    return 0;
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 6740 KB Output is correct

Subtask #2
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 68 ms 14828 KB Output is correct
2 Correct 123 ms 25064 KB Output is correct

Subtask #3
70.0 / 70.0

# Verdict Execution time Memory Grader output
1 Correct 229 ms 50572 KB Output is correct
2 Correct 402 ms 75600 KB Output is correct
3 Correct 407 ms 85076 KB Output is correct
4 Correct 588 ms 110816 KB Output is correct
5 Correct 991 ms 171600 KB Output is correct
6 Correct 1310 ms 193316 KB Output is correct
7 Correct 1166 ms 196176 KB Output is correct