Submission #631756

#	Time	Username	Problem	Language	Result	Execution time	Memory
631756		ZsofiaKeresztely	Kpart (eJOI21_kpart)	C++14	100 / 100	720 ms	660 KiB

Main

#include <bits/stdc++.h>
using namespace std;
 
int main(){
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
  int t, n, s, sum, op;
  cin >> t;
  while (t--){
    cin >> n;
    int a[n + 1];
    bool k[n + 1];
    s = a[0] = k[0] = 0;
    for (int i=1; i<n+1; i++){
      cin >> a[i];
      s += a[i];
      k[i] = true;
    }
    s /= 2;
    int lasti[s + 1] = {};
    for (int i=1; i<n+1; i++){
      for (int j=s-a[i]; j>0; j--){
        lasti[j + a[i]] = max(lasti[j + a[i]], lasti[j]);
      }
      if (a[i] <= s){
        lasti[a[i]] = i;
      }
      sum = 0;
      for (int j=i; j>0; j--){
        sum += a[j];
        if (sum % 2 || lasti[sum / 2] < j){
          k[i - j + 1] = false;
        }
      }
    }
    op = 0;
    for (int x : k){
      if (x){
        op++;
      }
    }
    cout << op << " ";
    for (int i=1; i<n+1; i++){
      if (k[i]){
        cout << i << " ";
      }
    }
    cout << "\n";
  }
  return 0;
}

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