Submission #167971

#	Time	Username	Problem	Language	Result	Execution time	Memory
167971		juggernaut	Bank (IZhO14_bank)	C++14	25 / 100	1054 ms	504 KiB

bank

#include<bits/stdc++.h>
using namespace std;
int n,m,i,mask,mx,a[11],b[10],sum,j;
bool dp[11][1024];
main(){
    scanf("%d%d",&n,&m);
    mx=(1<<m)-1;
    for(i=0;i<=mx;i++)dp[0][i]=1;
    for(i=1;i<=n;i++)scanf("%d",&a[i]);
    for(i=0;i<m;i++)scanf("%d",&b[i]);
  	if(n==1){
      for(mask=0;mask<=mx;mask++){
        sum=0;
        for(i=0;i<m;i++)if(mask&(1<<i))sum+=b[i];
        if(sum==a[1]){
          puts("YES");
          return 0;
        }
      }
        puts("NO");
      return 0;
    }
    for(i=1;i<=n;i++){
        for(mask=0;mask<=mx;mask++){
            sum=0;
            for(j=0;j<m;j++)
                if(mask&(1<<j))sum+=b[j];
            if(sum==a[i]){
                for(j=0;j<=mx;j++)
                    if((j&(mx^mask))==j&&dp[i-1][j]){
                        dp[i][(mask|j)]=1;
                        break;
                    }
            }
        }
    }
    for(i=0;i<=mx;i++)if(dp[n][i]){
        puts("YES");
        return 0;
    }
    puts("NO");
}

Compilation message (stderr)

bank.cpp:5:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main(){
      ^
bank.cpp: In function 'int main()':
bank.cpp:6:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d",&n,&m);
     ~~~~~^~~~~~~~~~~~~~
bank.cpp:9:27: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     for(i=1;i<=n;i++)scanf("%d",&a[i]);
                      ~~~~~^~~~~~~~~~~~
bank.cpp:10:26: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     for(i=0;i<m;i++)scanf("%d",&b[i]);
                     ~~~~~^~~~~~~~~~~~

• Subtask 1: 0 / 19
• Subtask 2: 25 / 25
• Subtask 3: 0 / 27
• Subtask 4: 0 / 29