Submission #459601

#	Time	Username	Problem	Language	Result	Execution time	Memory
459601		JovanB	Bootfall (IZhO17_bootfall)	C++17	100 / 100	804 ms	6248 KiB

bootfall

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

using ll = long long;
using ld = long double;

const int N = 500;
const int ofs = N*N;
const int MOD = 1000000007;

int add(int a, int b){ a += b; if(a >= MOD) a -= MOD; return a; }
int sub(int a, int b){ return add(a, MOD - b); }

int moze[N*N+5];
int dp[2*N*N+5];
int ddp[2*N*N+5];
int a[N+5];

int main(){
    ios_base::sync_with_stdio(false), cin.tie(0);
    cout.precision(10);
    cout << fixed;

    int n;
    cin >> n;
    int poc = 0;
    for(int i=1; i<=n; i++){
        cin >> a[i];
        poc += a[i];
        a[i] *= 2;
    }
    dp[ofs - poc] = 1;
    for(int i=1; i<=n; i++){
        for(int j=poc; j>=-poc+a[i]; j--){
            dp[ofs + j] = add(dp[ofs + j], dp[ofs + j - a[i]]);
        }
    }
    if(dp[ofs] == 0){
        cout << "0\n";
        return 0;
    }
    for(int i=1; i<=n; i++){
        for(int j=-poc; j<=poc; j++){
            ddp[ofs + j] = dp[ofs + j];
            if(ofs + j - a[i] >= 0) ddp[ofs + j] = sub(ddp[ofs + j], ddp[ofs + j - a[i]]);
        }
        for(int j=1; j<=poc; j++) moze[j] += (ddp[ofs + j - a[i]/2] > 0);
    }
    vector <int> vec;
    for(int i=1; i<=poc; i++) if(moze[i] == n) vec.push_back(i);
    cout << vec.size() << "\n";
    for(auto c : vec) cout << c << " ";
    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: 35 / 35