제출 #468422

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
468422		four_specks	은행 (IZhO14_bank)	C++17	100 / 100	135 ms	8576 KiB

bank

#include <bits/stdc++.h>

using namespace std;

void solve()
{
    int n;
    int m;
    cin >> n >> m;

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

    bool possible = 0;

    vector<pair<int, int>> dp(1 << m, pair(-1, 0));
    dp[0] = pair(0, 0);
    for (int mask = 0; mask < 1 << m && !possible; mask++)
    {
        for (int j = 0; j < m; j++)
        {
            if ((mask >> j) & 1)
            {
                auto tr = dp[mask ^ (1 << j)];
                if (tr.first != -1)
                {
                    if (tr.second + b[j] == a[tr.first])
                        tr.first++, tr.second = 0;
                    else if (tr.second + b[j] < a[tr.first])
                        tr.second += b[j];
                    else
                        tr.first = -1, tr.second = 0;

                    dp[mask] = max(dp[mask], tr);
                }
            }
        }

        possible |= dp[mask].first == n;
    }

    cout << (possible ? "YES" : "NO") << '\n';
}

int main()
{
    ios_base::sync_with_stdio(false), cin.tie(NULL);

    solve();

    return 0;
}

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