Submission #767181

#	Time	Username	Problem	Language	Result	Execution time	Memory
767181		Trunkty	Bootfall (IZhO17_bootfall)	C++14	65 / 100	1086 ms	19160 KiB

bootfall

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define int ll

int n,tot,mod=1e9+7;
int arr[505],dp[250005],curr[250005];
bitset<250005> bs[505];

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n;
    for(int i=1;i<=n;i++){
        cin >> arr[i];
        tot += arr[i];
    }
    dp[0] = 1;
    for(int i=1;i<=n;i++){
        for(int j=250000;j>=arr[i];j--){
            dp[j] += dp[j-arr[i]];
            dp[j] %= mod;
        }
    }
    if(tot%2 or dp[tot/2]==0){
        cout << 0 << "\n";
        cout << "\n";
        return 0;
    }
    for(int j=1;j<=n;j++){
        for(int i=0;i<=250000;i++){
            curr[i] = dp[i];
            if(i>=arr[j]){
                curr[i] -= curr[i-arr[j]];
                curr[i] += mod;
                curr[i] %= mod;
            }
            if(curr[i]){
                bs[j][i] = true;
            }
        }
    }
    vector<int> ans;
    for(int i=1;i<=250000;i++){
        bool yes = true;
        for(int j=1;j<=n;j++){
            if((tot-arr[j]+i)%2){
                yes = false;
                break;
            }
            int x = (tot-arr[j]+i)/2;
            if(!dp[x]){
                yes = false;
                break;
            }
            if(!bs[j][x]){
                yes = false;
                break;
            }
        }
        if(yes){
            ans.push_back(i);
        }
    }
    cout << ans.size() << "\n";
    for(int i:ans){
        cout << i << " ";
    }
    cout << "\n";
    return 0;
}

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