제출 #1352001

#제출 시각아이디문제언어결과실행 시간메모리
1352001bbbirosBank (IZhO14_bank)C++20
71 / 100
1096 ms44572 KiB
#include <iostream>
#include <vector>
#include <cstring>
#include <string>
#include <map>
#define endl '\n'
using namespace std;

void speed()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
}
short p[20];
short b[20];
short n, m;
void read()
{
    cin >> n >> m;
    for (short i = 0; i < n; i++)
    {
        cin >> p[i];
    }
    for (short i = 0; i < m; i++)
    {
        cin >> b[i];
    }
}
const short MAXN = 20100;
vector<short> dp[MAXN + 5];
struct state
{
    short sum, i;
    bool used[20];
    state()
    {
        sum = i = 0;
        for(int i=0;i<m;i++)used[i]=0;
    }
    bool operator<(const state &x) const
    {
        if (i != x.i)
            return i < x.i;
        if (sum != x.sum)
            return sum < x.sum;
        for (short i = 0; i < 20; i++)
        {
            if (used[i] != x.used[i])
                return used[i] < x.used[i];
        }
        return false;
    }
    state operator+(short&idx)
    {
        state res;
        res.i = i;
        res.sum = sum-b[idx];
        for(int i=0;i<m;i++)
        {
            res.used[i]=used[i];
        }
        res.used[idx] = 1;
        return res;
    }
};
map<state, bool> s;
bool rec(state x)
{
    if (s.count(x))
        return s[x];
    if (x.i == n)
    {
        s[x] = 1;
        return true;
    }
    if (x.sum == 0)
    {
        state ne = x;
        ne.i = x.i + 1;
        ne.sum = p[ne.i];
        bool res = rec(ne);
        s[x] = res;
        return res;
    }
    for (short&p : dp[x.sum])
    {
        if (!x.used[p])
        {
            if (rec(x + p))
            {
                s[x]=1;
                return true;
            }
        }
    }
    s[x]=0;
    return false;
}
signed main()
{
    speed();
    read();
    dp[0].push_back(-1);
    for (short i = 0; i < m; i++)
    {
        for (short j = MAXN; j >= 0; j--)
        {
            if (dp[j].size())
            {
                dp[j + b[i]].push_back(i);
            }
        }
    }
    state x;
    x.i = 0;
    x.sum = p[0];
    if (rec(x))
    {
        cout << "YES" << endl;
    }
    else
    {
        cout << "NO" << endl;
    }
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...