Submission #92828

#	Time	Username	Problem	Language	Result	Execution time	Memory
92828		SamAnd	Bootfall (IZhO17_bootfall)	C++17	100 / 100	620 ms	124876 KiB

bootfall

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

int n;
int a[N];
int m;

int u[N * N];
int uu[N * N];
bool t[N][N * N];

int main()
{
    //freopen("input2.txt", "r", stdin);
    cin >> n;
    for (int i = 1; i <= n; ++i)
    {
        cin >> a[i];
        m += a[i];
    }

    u[0] = 1;

    for (int i = 1; i <= n; ++i)
    {
        for (int j = m; j >= 0; --j)
        {
            if (j + a[i] <= m)
                u[j + a[i]] += u[j];
        }
    }

    for (int i = 1; i <= n; ++i)
    {
        memcpy(uu, u, sizeof u);
        for (int j = 0; j <= m; ++j)
        {
            if (j + a[i] <= m)
                uu[j + a[i]] -= uu[j];
        }
        for (int j = 0; j <= m; ++j)
        {
            if (uu[j])
                t[i][j] = true;
        }
    }

    if ((m % 2) == 1 || !u[m / 2])
    {
        cout << 0 << endl;
        return 0;
    }

    vector<int> ans;
    for (int x = 1; x <= m; ++x)
    {
        bool z = true;
        for (int i = 1; i <= n; ++i)
        {
            int y = (m - a[i] + x) / 2;
            if (!((m - a[i] + x) % 2 == 0 && y <= m && t[i][y]))
            {
                z = false;
                break;
            }
        }
        if (z)
            ans.push_back(x);
    }

    cout << ans.size() << endl;
    for (int i = 0; i < ans.size(); ++i)
        cout << ans[i] << ' ';
    cout << endl;

    return 0;
}

Compilation message (stderr)

bootfall.cpp: In function 'int main()':
bootfall.cpp:73:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < ans.size(); ++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