Submission #340566

#	Time	Username	Problem	Language	Result	Execution time	Memory
340566		Sprdalo	은행 (IZhO14_bank)	C++17	71 / 100	1083 ms	131952 KiB

bank

#include <bits/stdc++.h>

using namespace std;

int a[21], b[21], cnt[1<<20+5];
int n, m;

void ye(){
    cout << "YES\n";
    exit(0);
}

void solve(int x, int suma, int mask, int j){
    if (x>n) ye();

    if (!suma){
        solve(x+1,a[x+1], mask,0);
        return;
    }

    int cn = cnt[mask];
    if (m-cn-1 < x-n) return;
    for (int i = j; i < m; ++i){
        if (m-cn-1 < x-n) break;
        if ((mask&(1 << i)) || b[i] > suma) continue;
        solve(x, suma-b[i], mask^(1<<i),i+1); 
    }
}

signed main()
{
    ios_base::sync_with_stdio(false); 
    cin.tie(nullptr); 
    cout.tie(nullptr); 
    cerr.tie(nullptr);    

    for (int i = 0; i < 1<<20+5; ++i)
        cnt[i]=__builtin_popcount(i);

    cin >> n >> m;

    for (int i = 1; i <= n; ++i)
        cin >> a[i];
    for (int i = 0; i < m; ++i)
        cin >> b[i];

    solve(1, a[1], 0, 0);

    cout << "NO\n";
}

Compilation message (stderr)

bank.cpp:5:28: warning: suggest parentheses around '+' inside '<<' [-Wparentheses]
    5 | int a[21], b[21], cnt[1<<20+5];
      |                          ~~^~
bank.cpp: In function 'int main()':
bank.cpp:37:30: warning: suggest parentheses around '+' inside '<<' [-Wparentheses]
   37 |     for (int i = 0; i < 1<<20+5; ++i)
      |                            ~~^~

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