Submission #934013

#	Time	Username	Problem	Language	Result	Execution time	Memory
934013		dilanyan	Bootfall (IZhO17_bootfall)	C++17	100 / 100	196 ms	5844 KiB

bootfall

//-------------dilanyan------------\\ 
 
#define _CRT_SECURE_NO_WARNINGS
#include<bits/stdc++.h>
#include<stdio.h>
using namespace std;
 
//------------------Kargpefines--------------------\\ 
 
#define ll long long
#define pb push_back
#define all(u) (u).begin(), (u).end()
#define pqueue priority_queue
#define upper upper_bound
#define lower lower_bound
#define umap unordered_map
#define uset unordered_set
 
void KarginSet(string name = "") {
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    if (name.size()) {
        freopen((name + ".in").c_str(), "r", stdin);
        freopen((name + ".out").c_str(), "w", stdout);
    }
}
 
//-------------------KarginConstants------------------\\ 
 
const ll mod = 1000000007;
const ll inf = 1e9;
 
//-------------------KarginCode-----------------------\\ 
 
const int N = 300005;
int a[N], dp[N], dp1[N];
int good[N];

void KarginSolve() {
    int n; cin >> n;
    int sum = 0;
    for (int i = 1;i <= n;i++) {
        cin >> a[i];
        sum += a[i];
    }
    dp[0] = 1;
    for (int i = 1;i <= n;i++) {
        for (int j = sum - a[i];j >= 0;j--) {
            if (dp[j]) dp[j + a[i]] += dp[j];
        }
    }
    if (sum % 2 || !dp[sum / 2]) {
        cout << "0\n";
        return;
    }
    for (int i = 1;i <= n;i++) {
        for (int j = 0;j <= sum;j++) dp1[j] = dp[j];
        sum -= a[i];
        for (int j = 0;j <= sum - a[i];j++) dp1[a[i] + j] -= dp1[j];
        for (int j = sum / 2;j <= sum;j++) {
            if (dp1[j]) good[2 * j - sum]++;
        }
        sum += a[i];
    }
    vector<int> ans;
    for (int i = 1;i <= sum;i++) {
        if (good[i] == n) ans.pb(i);
    }
    cout << ans.size() << '\n';
    for (int i : ans) cout << i << ' ';
}

int main() {
    KarginSet();
    int test = 1;
    //cin >> test;
    while (test--) {
        KarginSolve();
    }
    return 0;
}

Compilation message (stderr)

bootfall.cpp:1:1: warning: multi-line comment [-Wcomment]
    1 | //-------------dilanyan------------\\
      | ^
bootfall.cpp:8:1: warning: multi-line comment [-Wcomment]
    8 | //------------------Kargpefines--------------------\\
      | ^
bootfall.cpp:27:1: warning: multi-line comment [-Wcomment]
   27 | //-------------------KarginConstants------------------\\
      | ^
bootfall.cpp:32:1: warning: multi-line comment [-Wcomment]
   32 | //-------------------KarginCode-----------------------\\
      | ^
bootfall.cpp: In function 'void KarginSet(std::string)':
bootfall.cpp:22:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   22 |         freopen((name + ".in").c_str(), "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
bootfall.cpp:23:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   23 |         freopen((name + ".out").c_str(), "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 6 / 6
• Subtask 2: 7 / 7
• Subtask 3: 15 / 15
• Subtask 4: 16 / 16
• Subtask 5: 21 / 21
• Subtask 6: 35 / 35