Submission #1107377

#TimeUsernameProblemLanguageResultExecution timeMemory
1107377vjudge1Bank (IZhO14_bank)C++17
71 / 100
1067 ms258404 KiB
#include <bits/stdc++.h>
#pragma optimize("g", on)
#pragma GCC optimize ("inline")
#pragma GCC optimize ("Ofast")
#pragma GCC optimize ("unroll-loops")
#pragma GCC optimize ("03")
#pragma GCC target ("sse,sse2,sse3,ssse3,sse4,popcnt,abm,avx2,mmx,fma,avx,tune=native")
using namespace std;
const int N = 2e5+ 4;
int n , T , q , m ;
signed main(){
    cin>> n >> m ;
    int a[n+1] , b[m] , ok =0 ;
    for(int i= 1 ; i <= n ;i++)cin>> a[i];
    for(int i =0 ; i < m; i++)cin>> b[i];
    vector<vector<int>>mp[25000];
        for(int j =  0 ; j < (1 << m )- 1; j++){
            int x =0 ;
            vector<int>v;
            for(int u =0 ; u < m ; u++){
                if((j >> u )&1){
                    x+=b[u];
                    v.push_back(b[u]);
                }
            }
            mp[x].push_back(v);
        }
    queue<pair<int , map<int, int >>>q;
    map<int ,int >h;
    for(int i =0  ; i < m ; i++){
        h[b[i]]++;
    }
    q.push({1 , h });
    while(q.size()){
        pair<int , map<int , int >> p= q.front();
        q.pop();
        int v = p.first ;
        map<int ,int >us = p.second;
        for(auto it:mp[a[v]]){
                int ko =0 ;
            for(auto at:it){
                if(us[at])ko++;
                us[at]--;
            }
            if(ko== it.size()){
                if(v == n ){
                    cout << "YES";
                    return 0 ;
                }
                q.push({v + 1 ,us});
            }
            for(auto at:it){
                    us[at]++;
                }
        }
    }
    cout << "NO";

}

Compilation message (stderr)

bank.cpp:2: warning: ignoring '#pragma optimize ' [-Wunknown-pragmas]
    2 | #pragma optimize("g", on)
      | 
bank.cpp: In function 'int main()':
bank.cpp:45:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |             if(ko== it.size()){
      |                ~~^~~~~~~~~~~~
bank.cpp:13:25: warning: unused variable 'ok' [-Wunused-variable]
   13 |     int a[n+1] , b[m] , ok =0 ;
      |                         ^~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...