Submission #566421

#	Time	Username	Problem	Language	Result	Execution time	Memory
566421		RealSnake	Bootfall (IZhO17_bootfall)	C++14	13 / 100	1065 ms	496 KiB

bootfall

#include "bits/stdc++.h"
using namespace std;

#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;

#define ll long long
#define mod 1000000007

ofstream fout(".out");
ifstream fin(".in");

int n;
ll arr[501], dp[250001];
ll sum;

void remove(ll x) {
    for(int i = x; i <= sum; i++)
        dp[i] -= dp[i - x];
}

void add(ll x) {
    for(int i = sum; i >= x; i--)
        dp[i] += dp[i - x];
}

signed main() {

    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n;
    for(int i = 0; i < n; i++) {
        cin >> arr[i];
        sum += arr[i];
    }
    dp[0] = 1;
    for(int i = 0; i < n; i++) {
        for(int j = sum; j >= arr[i]; j--)
            dp[j] += dp[j - arr[i]];
    }
    if(sum % 2 || !dp[sum / 2]) {
        cout << 0;
        return 0;
    }
    vector<int> ans;
    for(int i = 1; i <= sum; i++) {
        bool bb = 1;
        for(int j = 0; j < n; j++) {
            ll x = sum - arr[j] + i;
            remove(arr[j]);
            add(x);
            if(i % 2 != arr[j] % 2 || !dp[x / 2])
                bb = 0;
            remove(x);
            add(arr[j]);
            if(!bb)
                break;
        }
        if(bb)
            ans.push_back(i);
    }
    cout << ans.size() << "\n";
    for(int i : ans)
        cout << i << " ";

    return 0;
}

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