oj.uz

Submission #978850

# Submission time Handle Problem Language Result Execution time Memory
978850 2024-05-09T19:38:03 Z VMaksimoski008 Kpart (eJOI21_kpart) C++17
10 / 100
 2000 ms 348 KB 
#include <bits/stdc++.h>

#define sz(x) (int)x.size()
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
//#define int long long

using namespace std;

using ll = long long;
using ull = unsigned long long;
using ld = long double;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

const int mod = 1e9 + 7;
const int LOG = 20;
const int maxn = 1e5 + 5;
const double eps = 1e-9;

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

    vector<int> v(n+1), pref(n+1);
    for(int i=1; i<=n; i++) {
        cin >> v[i];
        pref[i] = pref[i-1] + v[i];
    }

    vector<bool> ans(n+1, 1);
    //bitset<100001> bs;

    for(int i=1; i<=n; i++) {
        for(int j=i; j<=n; j++) {
            if((pref[j] - pref[i-1]) % 2 == 1) {
                ans[j-i+1] = 0;
                continue;
            }

            if(!ans[j-i+1]) continue;

            bitset<100001> bs;
            bs[0] = 1;
            for(int k=i; k<=j; k++) bs |= (bs << v[k]);

            if(!bs[(pref[j]-pref[i-1])/2]) {
                ans[j-i+1] = 0;
            }
        }
    }

    vector<int> res;
    for(int i=1; i<=n; i++) if(ans[i]) res.push_back(i);
    cout << res.size() << " ";
    for(int &x : res) cout << x << " ";
    cout << '\n';
}

int32_t main() {
    int t = 1; cin >> t;
    while(t--) solve();

    return 0;
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 344 KB Output is correct

Subtask #2
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 628 ms 344 KB Output is correct
2 Execution timed out 2058 ms 348 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #3
0 / 70.0

# Verdict Execution time Memory Grader output
1 Execution timed out 2057 ms 348 KB Time limit exceeded
2 Halted 0 ms 0 KB -