Submission #1176591

#TimeUsernameProblemLanguageResultExecution timeMemory
1176591giabao249Bank (IZhO14_bank)C++20
100 / 100
479 ms30264 KiB
#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 s = 0;
        FOR(i , 0 , m){
            if(bit(i) & mask) s += cost[i];
        }
        sum[mask] = s;
    }
    FOR(mask , 0 , bit(m)) f[0][mask] = 1;
    FOR(i, 0, n) {
        FOR(mask, 0, bit(m)) {
            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:67:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   67 |         freopen(in.c_str(), "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...