Submission #1163298

#	Time	Username	Problem	Language	Result	Execution time	Memory
1163298		lopkus	Kpart (eJOI21_kpart)	C++20	10 / 100	2095 ms	472 KiB

Main

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 100001;

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int tt;
  cin >> tt;
  while(tt--) {
    int n;
    cin >> n;
    vector<int> a(n + 1);
    for(int i = 1; i <= n; i++) {
      cin >> a[i];
    }
    function<int(int, int)> can = [&](int l, int r) {
      int sum = 0;
      for(int i = l; i <= r; i++) {
        sum += a[i];
      }
      if(sum % 2) {
        return 0;
      }
      bitset<MAXN> b;
      b[0] = 1;
      for(int i = l; i <= r; i++) {
        b |= (b << a[i]);
      }
      int w = b[sum / 2];
      return w;
    };
    vector<int> ans;
    for(int k = 1; k <= n; k++) {
      int ok = 1;
      for(int i = 1; i + k - 1 <= n; i++) {
        ok &= can(i, i + k - 1);
      }
      if(ok) {
        ans.push_back(k);
      }
    }
    cout << ans.size() << " ";
    for(auto x : ans) {
      cout << x << " ";
    }
    cout << "\n";
  }
}

• Subtask 1: 10 / 10
• Subtask 2: 0 / 20
• Subtask 3: 0 / 70