Submission #1279751

#	Time	Username	Problem	Language	Result	Execution time	Memory
1279751		nhmktu	은행 (IZhO14_bank)	C++17	0 / 100	1096 ms	4420 KiB

bank

#include <bits/stdc++.h>
#define ll long long
#define FOR(i,a,b) for(int i=a;i<=b;i++)
#define ROF(i,a,b) for(int i=a;i>=b;i--)
#define pi pair<int,int>
#define pii pair<int,pi>
#define fi first
#define se second
#define pb push_back
#define all(x) x.begin(), x.end()
#define sz(a) (int) a.size()
#define endl '\n'
#define data "bank"

using namespace std;
const ll linf = 1e18;
const int inf = 1e9;
const int MOD = 1e9 + 7, N = 20;

void add(int &a, int b)
{
    a += b;
    if(a>=MOD)
        a-=MOD;
    if(a<0)
        a += MOD;
}

int modulo(int x)
{
    if(x<=1)
        return 1;
    return (MOD - MOD/x) * modulo(MOD/x) % MOD;
}

int mul(int a, int b)
{
    return (1ll *a%MOD * b%MOD) % MOD;
}

int n, m;
int a[N+3];
int b[N+3];
int dp[2][(1 << N) + 3];

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

void solve(void)
{
    FOR(i, 1, n)
    {
        int cur = (i & 1);
        int pre = !cur;

        FOR(msk, 0, (1 << m) -1) dp[cur][msk] = 0;

        FOR(msk, 0, (1 << m) -1)
        {
            for(int submsk = msk; submsk; submsk = (submsk - 1) & msk)
            {
                int subset = submsk ^ msk;
                int tolt = 0;
                FOR(j, 0, m - 1)
                {
                    if(subset >> j & 1) tolt += b[j+1];
                }
                if(tolt == a[i])
                    dp[cur][msk] = max(dp[cur][msk], dp[pre][submsk] + 1);
            }
        }
    }

    bool ok = false;
    FOR(msk, 0, (1 << m) -1)
        if(dp[n & 1][msk] == n) ok = true;
    if(ok) cout << "YES";
    else cout << "NO";
}

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

    if(fopen(data".inp", "r"))
    {
        freopen(data".inp","r",stdin);
        freopen(data".out","w",stdout);
    }

    inp();
    solve();
    return 0;
}

Compilation message (stderr)

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

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