Submission #880138

#	Time	Username	Problem	Language	Result	Execution time	Memory
880138		boris_7	은행 (IZhO14_bank)	C++17	19 / 100	70 ms	6996 KiB

bank

#include<bits/stdc++.h>

using namespace std;
using ll = long long;


void solve(){
    int n,m;
    cin>>n>>m;
    vector<int>a(n);
    vector<int>b(m);
    for(int &i:a){
        cin>>i;
    }
    for(int &i:b){
        cin>>i;
    }
    vector<vector<int>>sum(1001);
    for(int i = 0;i<(1<<m);i++){
        int gumar = 0;
        for(int j = 0;j<m;j++){
            if(i&(1<<j)){
                gumar+=b[j];
            }
        }
        if(gumar<=1000){
            sum[gumar].push_back(i);
        }
    }
    vector<bool>last((1<<m)+1),cur((1<<m)+1);
    for(int i = 0;i<int(sum[a[0]].size());i++){
        last[sum[a[0]][i]]=1;
    }
    for(int i = 1;i<n;i++){
        for(int j = 0;j<(1<<m);j++){
            for(int &k:sum[a[i]]){
                if(last[j]==0) continue;
                if((k&j)==0){
                    cur[j+k]=1;
                }
            }
        }
        last = cur;
        for(bool c:cur) c = 0;
    }
    int ans = 0;
    for(bool i:last){
        ans |= int(i);
    }
    if(ans){
        cout<<"YES";
    }
    else{
        cout<<"NO";
    }
}

signed main(){
    ios_base::sync_with_stdio(0);
    cin.tie(nullptr);
    // ll t;cin>>t;while(t--)
        solve();
}

Compilation message (stderr)

bank.cpp: In function 'void solve()':
bank.cpp:44:18: warning: variable 'c' set but not used [-Wunused-but-set-variable]
   44 |         for(bool c:cur) c = 0;
      |                  ^

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