Submission #1084656

#	Time	Username	Problem	Language	Result	Execution time	Memory
1084656		lhlephuocdao	Bank (IZhO14_bank)	C++14	44 / 100	559 ms	262144 KiB

bank

#include <bits/stdc++.h>
 
using namespace std;
 
const int Z = 20;
int N, M, A[Z], B[Z];
vector<int> S[Z];
vector<int> dp[1<<Z]; // dp[s] : states of subset S

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    cin >> N >> M;
    for (int i = 0; i < N; i++) cin >> A[i];
    for (int i = 0; i < M; i++) cin >> B[i];
 
	for (int state = 1; state < (1 << M); state++) {
		int sum = 0;
		for (int i = 0; i < M; i++) if (state & (1 << i)) sum += B[i];
		for (int i = 0; i < N; i++) if (sum == A[i]) S[i].push_back(state);
	}
 
    //dp[0] = empty
    for (int s = 1; s < (1<<N); s++)
    {
        for (int i = 0; i < N; i++)
        {
            if (!(s & (1<<i))) continue;

            if (dp[s^(1<<i)].empty()) {
                if (__builtin_popcount(s) == 1) dp[s] = S[i];
                continue;
            }

            for (auto cur_state : S[i])
                for (auto prev_state : dp[s^(1<<i)])
                {
                    if (cur_state & prev_state) continue;
                    dp[s].push_back(cur_state | prev_state);
                }
        }
    }

    if (dp[(1<<N)-1].empty())
        cout << "NO";
    else
        cout << "YES";
 
	return 0;
}

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:47:5: warning: this 'else' clause does not guard... [-Wmisleading-indentation]
   47 |     else
      |     ^~~~
bank.cpp:50:2: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'else'
   50 |  return 0;
      |  ^~~~~~

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