This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#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++){
bool f = 0;
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);){
if(nmask & mask){
for(int j = 0;j<m;j++){
if((1&(nmask>>j)) && (1&(mask>>j))){
nmask+=(1<<j);
continue;
}
}
}
for(auto x:mp[nmask]){
if(x == i-1){
f = 1;
mp[mask].pb(i);
break;
}
}
nmask++;
if(f) break;
}
}
}
if(!f){
no;
return;
}
}
yes;
}
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 time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |