Submission #1176586

#	Time	Username	Problem	Language	Result	Execution time	Memory
1176586		giabao249	Bank (IZhO14_bank)	C++20	19 / 100	14 ms	10484 KiB

bank

#include <bits/stdc++.h>
using namespace std;
using pii = pair<int,int>;
#define int long long
#define FOR(i , a , b) for(int i = a ; i < b ; i++)
#define REP(i , a , b) for(int i = a ; i <= b ; i++)
#define FORD(i , a , b) for(int i = a ; i >= b ; i--)
#define FORE(i, v) for (__typeof((v).begin()) i = (v).begin(); i != (v).end(); i++)
#define el '\n'
#define all(x) x.begin(), x.end()
#define bit(x) (1 << (x))
#define log(...) logger(#__VA_ARGS__, __VA_ARGS__)
const int inf = 1e18;
const int base = 311;
const int mod = 1e9 + 7;
template <typename... Args> void logger(string vars, Args&&... values);
template<class T> bool ckmin(T& a, const T& b)
{
    return b < a ? a = b, 1 : 0;
}
template<class T> bool ckmax(T& a, const T& b)
{
    return a < b ? a = b, 1 : 0;
}
//Think
const int N = 21;
string name = "Task";
bool f[N][bit(N)];
int sum[bit(N)] ;
int n, m;
int cost[N], staff[N];
vector<vector<int>> valid;
void GOTO_OLP2025(){
    cin >> n >> m;
    FOR(i, 0, n) cin >> staff[i];
    FOR(i, 0, m) cin >> cost[i];
    valid.resize(n);
    FOR(mask, 0, bit(m)) {
        int newMask = mask & (mask - 1);
        int lg = __lg(mask ^ newMask);
        sum[mask] = sum[newMask] + cost[lg];
    }
    FOR(mask , 0 , bit(m)) f[0][mask] = 1;
    FOR(mask, 0, bit(m)) {
        FOR(i, 0, n) {
            if(!f[i][mask]) continue;
            int notUsed = (bit(m) - 1) ^ mask;
            if(sum[notUsed] == staff[i]){
                for(int s = mask ; s ; s = (s - 1) & mask){
                    f[i + 1][s ^ notUsed] |= f[i][s];
                }
                f[i + 1][notUsed] |= f[i][0];
            }
        }
    }
    if(f[n][bit(m) - 1]) cout << "YES" << el;
    else cout << "NO" << el;
}

void file(string s)
{
    string in = s + ".in";
    string out = s + ".out";
    if(fopen(in.c_str(), "r")) {
        freopen(in.c_str(), "r", stdin);
    }
}
int32_t main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    file(name);
    //int t ; cin >> t;
    //while(t--)
    GOTO_OLP2025();
    return 0;
}
template <typename... Args>
void logger(string vars, Args&&... values)
{
    cerr << "[";
    stringstream ss(vars);
    string var;
    vector<string> varNames;
    while (getline(ss, var, ',')) {
        while (!var.empty() && var.front() == ' ') var.erase(0, 1);
        varNames.push_back(var);
    }
    string delim = "";
    size_t i = 0;
    ((cerr << delim << varNames[i++] << " = " << values, delim = ", "), ...);
    cerr << "]\n";
}

Compilation message (stderr)

bank.cpp: In function 'void file(std::string)':
bank.cpp:65:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   65 |         freopen(in.c_str(), "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

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