#include <bits/stdc++.h>
// iostream is too mainstream
#include <cstdio>
// bitch please
#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <list>
#include <cmath>
#include <iomanip>
#include <time.h>
#define dibs reserve
#define OVER9000 1234567890123456789LL
#define ALL_THE(CAKE,LIE) for(auto LIE =CAKE.begin(); LIE != CAKE.end(); LIE++)
#define tisic 47
#define soclose 1e-8
#define chocolate win
// so much chocolate
#define patkan 9
#define ff first
#define ss second
#define abs(x) ((x < 0)?-(x):x)
#define uint unsigned int
#define dbl long double
#define pi 3.14159265358979323846
using namespace std;
// mylittledoge
typedef long long cat;
#ifdef DONLINE_JUDGE
// palindromic tree is better than splay tree!
#define lld I64d
#endif
int cost_arr[2][1<<19];
int cost[1<<20];
int ans[1000000];
cat query[1000000];
cat CNT[20] = {};
void solve(int d, int B, int cl, int cr, int ql, int qr) {
CNT[d] += qr-ql;
if(ql == qr) return;
if(d == B) {
for(int i = ql; i < qr; i++)
ans[query[i]&((1LL<<20)-1)] += ((query[i]>>60) ? -cost[cl] : cost[cl]);
return;
}
int cnt[3] = {};
for(int i = ql; i < qr; i++) {
if((query[i]>>(40+d))&1) cnt[(query[i]>>(20+d))&1]++;
else cnt[2]++;
}
if(cnt[2] <= cnt[0] && cnt[2] <= cnt[1]) {
for(int i = 0; i < (1<<(B-1-d)); i++) {
cost_arr[0][i] = cost[cl+2*i];
cost_arr[1][i] = cost[cl+2*i+1];
}
for(int i = 0; i < (1<<(B-1-d)); i++) cost[cl+i] = cost_arr[0][i];
for(int i = 0; i < (1<<(B-1-d)); i++) cost[cl+(1<<(B-1-d))+i] = cost_arr[1][i];
int a = ql, b = ql+cnt[0]+cnt[2];
cat bits = (1LL<<(40+b)) + (1LL<<(20+b));
for(int i = ql; i < ql+cnt[0]+cnt[2]; i++) {
if((query[i]>>(40+d))&1) {
if((query[i]>>(20+d))&1) {
while(b < qr && (query[b]&bits) == bits) b++;
if(b == qr) continue;
swap(query[i], query[b]);
b++;
}
else {
while(a < ql+cnt[0] && ((query[a]>>(40+d))&1) == 1 && ((query[a]>>(20+d))&1) == 0) a++;
if(a >= i) continue;
swap(query[i], query[a]);
a++;
}
}
}
solve(d+1, B, cl, cl+(1<<(B-1-d)), ql, ql+cnt[0]+cnt[2]);
a = ql+cnt[0];
for(int i = ql; i < ql+cnt[0]; i++) if(((query[i]>>(40+d))&1) == 0) {
while(a < ql+cnt[0]+cnt[2] && ((query[a]>>(40+d))&1) == 0) a++;
swap(query[i], query[a]);
a++;
}
solve(d+1, B, cl+(1<<(B-1-d)), cr, qr-cnt[2]-cnt[1], qr);
return;
}
int x = (int) (cnt[1] < cnt[0]);
for(int i = 0; i < (1<<(B-1-d)); i++) {
cost_arr[0][i] = cost[cl+2*i+1-x];
cost_arr[1][i] = cost[cl+2*i] + cost[cl+2*i+1];
}
for(int i = 0; i < (1<<(B-1-d)); i++) cost[cl+i] = cost_arr[0][i];
for(int i = 0; i < (1<<(B-1-d)); i++) cost[cl+(1<<(B-1-d))+i] = cost_arr[1][i];
int a = ql, b = ql+cnt[1-x]+cnt[x];
cat bits = (1LL<<(40+d)) + (1-x) * (1LL<<(20+d));
for(int i = ql; i < ql+cnt[1-x]+cnt[x]; i++) {
if((query[i]>>(40+d))&1) {
if(((query[i]>>(20+d))&1) != x) {
while(a < ql+cnt[1-x] && (query[a]&bits) == bits) a++;
if(a >= i) continue;
swap(query[i], query[a]);
a++;
}
}
else {
while(b < qr && ((query[b-1]>>(40+d))&1) == 0) b++;
if(b == qr) continue;
swap(query[i], query[b]);
b++;
}
}
for(int i = ql+cnt[1-x]; i < ql+cnt[1-x]+cnt[x]; i++) query[i] ^= 1LL<<60;
solve(d+1, B, cl, cl+(1<<(B-1-d)), ql, ql+cnt[1-x]+cnt[x]);
a = ql+cnt[1-x];
for(int i = ql; i < ql+cnt[1-x]; i++) if((query[i]&bits) != bits) {
while(a < qr-cnt[2] && (query[a]&bits) == bits) a--;
if(a == qr-cnt[2]) break;
swap(query[i], query[a]);
a++;
}
for(int i = ql+cnt[1-x]; i < ql+cnt[1-x]+cnt[x]; i++) query[i] ^= 1LL<<60;
solve(d+1, B, cl+(1<<(B-1-d)), cr, qr-cnt[2]-cnt[x], qr);
}
int main() {
cin.sync_with_stdio(0);
cin.tie(0);
cout << fixed << setprecision(10);
int B, Q;
char buf[(1<<20) + 10];
scanf("%d %d %s", &B, &Q, buf);
for(int i = 0; i < (1<<B); i++) {
int b = 0;
for(int j = 0; j < B; j++) b = 2*b + ((i>>j)&1);
cost[b] = buf[i]-'0';
}
// query: mask, bits, id
char buf_small[30];
for(int i = 0; i < Q; i++) {
scanf("%s", buf_small);
query[i] = i;
for(int j = 0; j < B; j++) {
if(buf_small[j] == '?') continue;
query[i] += 1LL<<(j+40);
if(buf_small[j] == '1') query[i] += 1LL<<(j+20);
}
}
memset(ans, 0, sizeof(ans));
solve(0, B, 0, 1<<B, 0, Q);
for(int i = 0; i < Q; i++) cout << ans[i] << "\n";
// for(int i = 0; i <= B; i++) cerr << CNT[i] << " ";
// cerr << "\n";
return 0;}
// look at my code
// my code is amazing
Compilation message
snake_escaping.cpp: In function 'int main()':
snake_escaping.cpp:139:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d %d %s", &B, &Q, buf);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
snake_escaping.cpp:148:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%s", buf_small);
~~~~~^~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
5 ms |
4344 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
5 ms |
4344 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
5 ms |
4344 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
5 ms |
4344 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
5 ms |
4344 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
