Submission #474194

#	Time	Username	Problem	Language	Result	Execution time	Memory
474194		kingline	Bootfall (IZhO17_bootfall)	C++17	13 / 100	1082 ms	340 KiB

bootfall

#pragma GCC optimize ("Ofast")
#pragma GCC optimize ("unroll-loops")
#pragma GCC target("avx,avx2,fma")
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <iomanip>
#include <cmath>
#include <vector>
#include <string>
#include <stack>
#include <map>
#include <set>
#include <queue>
#define pb push_back
#define F first
#define mp make_pair
#define S second
#define pii pair <int, int >
#define pll pair <long long, long long >
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define all(data) data.begin() , data.end()
#define endl '\n'
#define bo cout << "OK" << endl;
#define file(data) freopen(data".in", "r", stdin); freopen(data".out", "w", stdout);
//freopen("nenokku_easy.in", "r", stdin);
//freopen("nenokku_easy.out", "w", stdout);
#define int long long

using namespace std;

typedef long double ld;
typedef long long ll;
typedef unsigned long long ull;

const ll N = 1e6 + 5;
const ll mod = 1e9 + 7;
const ll INF = 1e10;
const ll M = (1 << 17) + 5;
const ll LL = 1;
const ll p = 31;
bool ok;

ll q, n, m, k, a[N];
int dp[500 * 500], sum, mn = 501;
string s, s1, s2;

bool check2(vector < int > &v, int ign) {
    int sum1 = sum;
    sum += a[ign];
    if(sum1 % 2 == 0) {
        sum1 /= 2;
        dp[0] = 1;
        for(int i = 1; i <= sum1; i++) {
            dp[i] = 0;
        }
        for (int i = 0; i < v.size(); ++i) {
            if(i == ign) continue;
            for (int j = sum1; j >= 0; --j) {
                if (dp[j] == 1 && j + v[i] <= sum1) {
                    dp[j + v[i]] = 1;
                    if(j + v[i] == sum1) return 1;
                }
            }
        }
    }
    return 0;
}

vector < int > v;

bool check() {
    for(int i = 0; i <= n; i++) {
        sum -= a[i];
        if(!check2(v, i)) return 0;
    }
    return 1;
}

main()
{
    //file("B");
    ios;
    scanf("%d", &n);
    if(n % 2) {
        cout << 0;
        return 0;
    }
    v.pb(0);
    for(int i = 1; i <= n; i++) {
        //cin >> a[i];
        scanf("%d", &a[i]);
        mn = min(mn, a[i]);
        sum += a[i];
        v.pb(a[i]);
    }
    vector < int > ans;
    int right = sum - mn;
    for(int tmp = 1; tmp <= right; tmp++) {
        sum -= a[0];
        v[0] = a[0] = tmp;
        sum += a[0];
        if(check()) {
            ans.pb(tmp);
        }
    }
    printf("%d\n", ans.size());
    for(int i = 0; i < ans.size(); i++) {
        printf("%d ", ans[i]);
    }
}

Compilation message (stderr)

bootfall.cpp: In function 'bool check2(std::vector<long long int>&, long long int)':
bootfall.cpp:59:27: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |         for (int i = 0; i < v.size(); ++i) {
      |                         ~~^~~~~~~~~~
bootfall.cpp: At global scope:
bootfall.cpp:82:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   82 | main()
      | ^~~~
bootfall.cpp: In function 'int main()':
bootfall.cpp:86:13: warning: format '%d' expects argument of type 'int*', but argument 2 has type 'll*' {aka 'long long int*'} [-Wformat=]
   86 |     scanf("%d", &n);
      |            ~^   ~~
      |             |   |
      |             |   ll* {aka long long int*}
      |             int*
      |            %lld
bootfall.cpp:94:17: warning: format '%d' expects argument of type 'int*', but argument 2 has type 'll*' {aka 'long long int*'} [-Wformat=]
   94 |         scanf("%d", &a[i]);
      |                ~^   ~~~~~
      |                 |   |
      |                 |   ll* {aka long long int*}
      |                 int*
      |                %lld
bootfall.cpp:109:14: warning: format '%d' expects argument of type 'int', but argument 2 has type 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wformat=]
  109 |     printf("%d\n", ans.size());
      |             ~^     ~~~~~~~~~~
      |              |             |
      |              int           std::vector<long long int>::size_type {aka long unsigned int}
      |             %ld
bootfall.cpp:110:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  110 |     for(int i = 0; i < ans.size(); i++) {
      |                    ~~^~~~~~~~~~~~
bootfall.cpp:111:18: warning: format '%d' expects argument of type 'int', but argument 2 has type '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} [-Wformat=]
  111 |         printf("%d ", ans[i]);
      |                 ~^
      |                  |
      |                  int
      |                 %lld
bootfall.cpp:86:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   86 |     scanf("%d", &n);
      |     ~~~~~^~~~~~~~~~
bootfall.cpp:94:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   94 |         scanf("%d", &a[i]);
      |         ~~~~~^~~~~~~~~~~~~

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