Submission #1107477

#TimeUsernameProblemLanguageResultExecution timeMemory
1107477vjudge1Bank (IZhO14_bank)C++17
19 / 100
73 ms76620 KiB
#include <bits/stdc++.h>
#define yes cout<<"YES\n"
#define no cout<<"NO\n"
#define ff first
#define int long long
#define ss second
#define pb push_back
#define y1 zildjian

using namespace std;

const int N = 3e6+100;
const int INF = 1e18;
const int mod = 1e9+7;
const int dx[] = {-1,-1,0,1,1,1,0,-1,0};
const int dy[] = {0,1,1,1,0,-1,-1,-1,0};
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());

int binpow (int a, int n){
	if (n == 0)
		return 1;
	if (n % 2 == 1)
		return (binpow (a, n-1)%mod * a%mod)%mod;
	else {
		int b = binpow (a, n/2) % mod;
		return (b%mod * b%mod)%mod;
	}
}

int n,m;
int a[N],b[N],dp[N];
vector<int> mp[N];

void solve(){
    cin>>n>>m;
    for(int i = 1;i<=n;i++){
        cin>>a[i];
    }
    for(int i = 1;i<=m;i++){
        cin>>b[i];
    }   
    mp[0].pb(0);
    for(int i = 1;i<=n;i++){
        for(int mask = 0;mask<(1<<m);mask++){
            int sum = 0;
            int cnt = 0;
            for(int j = 1;j<=m;j++){
                if(1&(mask>>(j-1))){
                    sum+=b[j];
                    cnt++;
                }
            }
            if(sum == a[i]){
                for(int nmask = 0;nmask<(1<<(m-cnt));nmask++){
                    int kasm = 0;
                    int pw = 0;
                    for(int j = 0;j<m;j++){
                        if(!(1&(mask>>j))){
                            if(1&(nmask>>j)) kasm+=pow(2,pw);
                        }
                        pw++;
                    }
                    // cout<<nmask<<' '<<kasm<<' '<<mask<<'\n';
                    // 011
                    // 01010
                    // 00101
                    for(auto x:mp[kasm]){
                        if(x == i-1){
                            mp[mask].pb(i);
                            break;
                        }
                    }
                    if(mp[mask].size()){
                        if(i == n){
                            // cout<<mask<<' '<<nmask<<'\n';
                            yes;
                            return;
                        }
                        break;
                    }
                }
            }
        }
    }
    no;
}   

signed main(){
    // freopen("floyd.in","r",stdin); 
    // freopen("floyd.out","w",stdout);
    ios_base::sync_with_stdio(0);
    cin.tie(nullptr);
    // cout.tie(nullptr);
    int T = 1;
    // cin>>T;
    for(int i = 1;i<=T;i++){
        // cout<<"Case "<<i<<": ";
        solve();
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...