Submission #38792

#	Time	Username	Problem	Language	Result	Execution time	Memory
38792		DenXman111	Bootfall (IZhO17_bootfall)	C++14	100 / 100	423 ms	7920 KiB

bootfall

#include <bits/stdc++.h>
const int MAXN = 323456;
using namespace std;

int a[MAXN], ans[MAXN], dp[MAXN], d[MAXN];
int main()
{
//    freopen("bootfall.in", "r", stdin);
//    freopen("bootfall.out", "w", stdout);
    int n, i, j, sum = 0, x = 0;
    cin >> n;
    dp[0] = 1;
    for (i = 1; i <= n; i++){
        cin >> a[i];
        for (j = sum; j >= 0; j--) dp[j + a[i]] += dp[j];
        sum += a[i];
    }
    if (sum % 2 == 1 || dp[sum / 2] == 0) return cout << 0 << endl, 0;
    for (j = 0; j <= sum; j++) d[j] = dp[j];

    for (i = 1; i <= n; i++){
        for (j = 0; j <= sum; j++) {
            d[j + a[i]] -= d[j];
            x = abs(sum - j - a[i] - j);
            if (d[j] != 0 && ans[x] == i - 1) ans[x] = i;
            d[j] = dp[j];
        }
    }
    vector < int > v;
    for (i = 0; i <= sum; i++) if (ans[i] == n) v.push_back(i);
    cout << v.size() << endl;
    for (i = 0; i < v.size(); i++) cout << v[i] << " ";
    return 0;
}

Compilation message (stderr)

bootfall.cpp: In function 'int main()':
bootfall.cpp:32:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (i = 0; i < v.size(); i++) cout << v[i] << " ";
                   ^

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