Submission #38274

#	Time	Username	Problem	Language	Result	Execution time	Memory
38274		mirbek01	Bootfall (IZhO17_bootfall)	C++14	100 / 100	546 ms	5796 KiB

bootfall

# include <bits/stdc++.h>

# define pb push_back
# define fr first
# define sc second
# define mk make_pair

using namespace std;

const long long linf = 1e18 + 7;
const int inf = 1e9 + 7;
const int N = 250005;

typedef long long ll;

int n, a[N], dp[N], cnt[N], s;
vector <int> ans;

int main(){
      dp[0] = 1;

      scanf("%d", &n);

      for(int i = 1; i <= n; i ++){
            scanf("%d", &a[i]);
            s += a[i];
      }

      for(int i = 1; i <= n; i ++)
            for(int j = s; j - a[i] >= 0; j --)
                  dp[j] += dp[j - a[i]];

      if(s % 2 || !dp[s / 2])
      {
            cout << 0 << endl;
            return 0;
      }

      for(int i = 1; i <= n; i ++){
            int ss = s - a[i];

            for(int j = 0; j <= ss; j ++)
                  dp[j + a[i]] = (dp[j + a[i]] - dp[j]);

            for(int j = ss / 2 + 1; j <= ss; j ++)
                  if(dp[j]) cnt[j * 2 - ss] ++;

            for(int j = ss; j >= 0; j --)
                  dp[j + a[i]] += dp[j];
      }

      for(int i = 1; i <= s; i ++)
             if(cnt[i] == n) ans.pb(i);

      printf("%d\n", ans.size());

      for(int i : ans)
            printf("%d ", i);
}

Compilation message (stderr)

bootfall.cpp: In function 'int main()':
bootfall.cpp:55:32: warning: format '%d' expects argument of type 'int', but argument 2 has type 'std::vector<int>::size_type {aka long unsigned int}' [-Wformat=]
       printf("%d\n", ans.size());
                                ^
bootfall.cpp:22:22: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
       scanf("%d", &n);
                      ^
bootfall.cpp:25:31: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
             scanf("%d", &a[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