Submission #1210034

#	Time	Username	Problem	Language	Result	Execution time	Memory
1210034		nvc2k8	Bank (IZhO14_bank)	C++20	100 / 100	92 ms	8520 KiB

bank

#include <bits/stdc++.h>
#define TASK "kasdjkasd"
#define INT_LIM (int) 2147483647
#define LL_LIM (long long) 9223372036854775807
#define endl '\n'
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define BIT(i,x) (((i)>>(x))&1)
#define FOR(i,a,b) for(int i = (a); i<=(b); i++)
#define FORD(i,a,b) for(int i = (a); i>=(b); i--)
#define ll long long
#define pii pair<int,int>
using namespace std;
///------------------------------------------///
int n,m;
int a[21], b[21];
int pf[21];
int sum[1<<20];
int f[1<<20];

void inp()
{
    cin >> n >> m;
    FOR(i, 1, n)
    {
        cin >> a[i];
        pf[i] = pf[i-1]+a[i];
    }
    FOR(i, 1, m) cin >> b[i];
}

void solve()
{
    FOR(mask, 1, (1<<m)-1)
    {
        sum[mask] = b[__lg(mask)+1]+sum[mask^(1<<__lg(mask))];
    }

    bool flag = false;
    memset(f, -1, sizeof f);
    f[0] = 0;

    FOR(mask, 0, (1<<m)-1) if (f[mask]!=-1)
    {
        if (f[mask]==n)
        {
            flag = true; break;
        }
        int l = sum[mask]-pf[f[mask]];
        FOR(i, 1, m) if (!BIT(mask, i-1))
        {
//            cout << mask << ' ' << i << endl;
            if (l+b[i]<a[f[mask]+1])
            {
                f[mask^(1<<(i-1))] = max(f[mask^(1<<(i-1))], f[mask]);
            }
            else if (l+b[i]==a[f[mask]+1])
            {
                f[mask^(1<<(i-1))] = max(f[mask^(1<<(i-1))], f[mask]+1);
            }
        }
    }

    if (flag) cout << "YES";
    else cout << "NO";
}

signed main()
{
    ///--------------------------///
    ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    if (fopen(TASK".INP","r")!=NULL)
    {
        freopen(TASK".INP","r",stdin);
        freopen(TASK".OUT","w",stdout);
    }
    ///--------------------------///

    int NTEST = 1;
    //cin >> NTEST;

    while (NTEST--)
    {
        inp();
        solve();
    }

    return 0;
}

///------------------------------------------///

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:76:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   76 |         freopen(TASK".INP","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
bank.cpp:77:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   77 |         freopen(TASK".OUT","w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~

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