Submission #1107565

#	Time	Username	Problem	Language	Result	Execution time	Memory
1107565		the_ZHER	Bank (IZhO14_bank)	C++17	0 / 100	2 ms	336 KiB

bank

#include <bits/stdc++.h>
#define boost ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
const int N=2e5+5;
const int inf=1e9;
const int mod=1e9+7;
const int N1=1e6+1;
vector<int>v;
vector<int>v1;
struct edge{
    int v,time;
};
vector<int>dp[1001];
bool used[21][(1<<20)];
int n,m;
void dfs(int x,int msk){
    if(x==n+1){
        cout<<"YES\n";
        exit(0);
    }
    if (used[x][msk]) {
        return;
    }
    used[x][msk] = true;
    for(int i=0;i<dp[v[x]].size();i++){
        if((dp[v[x]][i]&msk)==0){
            dfs(x+1,msk+dp[v[x]][i]);
        }
    }
}
signed main(){
   freopen("bank.in", "r", stdin);
    freopen("bank.out", "w", stdout);
    boost;
    cin>>n>>m;
    v.push_back(0);
    v1.push_back(0);
    for(int i=0;i<n;i++){
        int x;
        cin>>x;
        v.push_back(x);
    }
    for(int i=0;i<m;i++){
        int x;
        cin>>x;
        v1.push_back(x);
    }
    for(int i=0;i<(1<<m);i++){
        int cnt=0;
        for(int j=0;j<m;j++){
            if(1&(i>>j)){
                cnt+=v1[j+1];
            }
        }
        if(cnt>1000){
            continue;
        }
        dp[cnt].push_back(i);
    }
    dfs(1,0);
    cout<<"NO\n";
}

Compilation message (stderr)

bank.cpp: In function 'void dfs(int, int)':
bank.cpp:26:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   26 |     for(int i=0;i<dp[v[x]].size();i++){
      |                 ~^~~~~~~~~~~~~~~~
bank.cpp: In function 'int main()':
bank.cpp:33:11: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |    freopen("bank.in", "r", stdin);
      |    ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:34:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   34 |     freopen("bank.out", "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

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