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>
using namespace std;
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) begin(a),end(a)
string s;
int n, l, q;
const int p3[] = {1,3,9,27,81,243,729,729*3,729*9,729*27,729*81,729*243,729*729};
int B1 = 6;
const int B2 = 20-6;
const int N = p3[6];
int score[N+2][(1<<B2)+2];
int calc(string lol){
int p = 1, ans = 0, xd;
for(int i = 0; i < sz(lol); i++){
if(lol[i]=='?') xd=2;
else xd = lol[i]-'0';
ans+=xd*p, p*=3;
}
return ans;
}
int calc(int lol){
int p = 1, ans = 0;
while(lol){
int r = lol%2; lol/=2;
ans+=r*p; p*=3;
}
return ans;
}
int32_t main(){
ios_base::sync_with_stdio(false); cin.tie(0);
cin >> l >> q >> s; n = (1<<l);/*
if(l<B1){
for(int _ = 0; _ < q; _++){
string t; cin >> t;
reverse(all(t));
int cnt = 0;
for(int i = 0; i < n; i++){
int ok = 1;
for(int j = 0; j < l; j++)
ok&=(t[j]=='?' or (t[j]-'0')==(i>>j&1));
cnt+=ok*(s[i]-'0');
}
cout << cnt << "\n";
}
return 0;
}*/
B1=min(B1,l-1); l-=B1;
for(int mask = 0; mask < n; mask++){
int mask1 = calc(mask&((1<<B1)-1));
int mask2 = mask>>B1;
int sc = s[mask]-'0';
score[mask1][mask2]+=sc;
}
for(int mask = 0; mask < (1<<(B1*2)); mask++){
string t = "", t2 = "";
for(int i = 0; i < B1*2; i+=2){
int cur = mask>>i&1;
t+=(char)(cur+'0');
if(mask>>(i+1)&1) cur=2;
t2+=(char)(cur+'0');
}
int mask1 = calc(t);
int mask2 = calc(t2);
if(t==t2) continue;
for(int i = 0; i < (1<<l); i++)
score[mask2][i]+=score[mask1][i];
}
for(int _ = 0; _ < q; _++){
string t; cin >> t; reverse(all(t));
int x = calc(t.substr(0,B1));
t = t.substr(B1);
int cnt = 0, ans = 0, xd2 = 0;
vector<int> pos, pos2;
pos.clear(); pos2.clear();
for(int i = 0; i < sz(t); i++){
if(t[i]=='?') pos.pb(i),cnt++;
else{
pos2.pb(i);
if(t[i]=='1') xd2+=1<<i;
}
}
if(cnt*2<=B2){
for(int mask = 0; mask < (1<<cnt); mask++){
int xd = 0;
for(int i = 0; i < cnt; i++)
if(mask>>i&1) xd+=1<<pos[i];
ans+=score[x][xd+xd2];
}
}
else{
cnt = sz(pos2);
for(int mask = 0; mask < (1<<cnt); mask++){
int num = __builtin_popcount(mask), xd=0;
for(int i = 0; i < cnt; i++)
if(mask>>i&1) xd+=(1<<pos2[i])*(t[pos2[i]]=='0');
ans+=(num%2?-1:1)*score[x][xd];
}
}
cout << ans << "\n";
}
}
# | 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... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |