Submission #91237

#	Time	Username	Problem	Language	Result	Execution time	Memory
91237		Vardanyan	Bootfall (IZhO17_bootfall)	C++14	100 / 100	977 ms	245172 KiB

bootfall

#include<bits/stdc++.h>
 
using namespace std;
const int N = 505;
bool dp[N][N*N*2];
int a[N];
int dp1[N*N*2];
int main()
{
    int n;
    scanf("%d",&n);
    int MX = 0;
    int sum = 0;
    for(int i = 1;i<=n;i++){
            scanf("%d",&a[i]);
            MX+=a[i];
            sum+=a[i];
    }
    sort(a+1,a+1+n);
        dp1[0] = 1;
        for(int j = 1;j<=n;j++){
            for(int s = 2*MX;s>=0;s--){
                if(dp1[s]) dp1[s+a[j]] += dp1[s];
            }
        }
    for(int i = 1;i<=n;i++){
        for(int s = 0;s<=2*MX;s++) dp1[s+a[i]] -= dp1[s];
        for(int s = 0;s<=2*MX;s++) dp[i][s] = dp1[s];
        for(int s = 2*MX;s>=0;s--) dp1[s+a[i]] += dp1[s];
    }
    if(!dp1[sum/2]){
        printf("0\n");
        return 0;
    }
    vector<int> ans;
    for(int x = 1;x<=MX;x++){
        sum+=x;
        bool F = true;
        for(int i = 1;i<=n;i++){
            sum-=a[i];
            int need = sum/2;
            sum+=a[i];
            if(need-x<0){
                F = false;
                break;
            }
            if(dp[i][need-x] && (sum-a[i])%2 == 0) continue;
            F = false;
            break;
        }
        sum-=x;
        if(F) ans.push_back(x);
    }
    printf("%d\n",ans.size());
    for(int i = 0;i<ans.size();i++) printf("%d ",ans[i]);
    printf("\n");
    return 0;
}

Compilation message (stderr)

bootfall.cpp: In function 'int main()':
bootfall.cpp:54:29: 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:55:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0;i<ans.size();i++) printf("%d ",ans[i]);
                   ~^~~~~~~~~~~
bootfall.cpp:11:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d",&n);
     ~~~~~^~~~~~~~~
bootfall.cpp:15:18: 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