Submission #1146699

#TimeUsernameProblemLanguageResultExecution timeMemory
1146699tntSnake Escaping (JOI18_snake_escaping)C++20
5 / 100
2096 ms18248 KiB
#include <bits/stdc++.h>
using namespace std;
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("Ofast,unroll-loops,fast-math,O3")

#define pb push_back                    
#define ll long long
#define int long long
//#define sort(all(v)) sort(v.begin(),v.end())

int mod = 998244353;
const int N = 1e5 + 10;
const int inf = 1e9;



signed main(){
    //freopen("mootube.in", "r", stdin);
    //freopen("mootube.out", "w", stdout);
    int l,q;
    cin >> l >> q;                                
    string s;                                      
    cin >> s;
    vector <ll> dp(1 << l),dp1(1 << l);
    for(int i = 0; i < (1 << l); i++){
    	dp[i] = dp1[i] = s[i] - '0';
    }
    for(int i = 0; i < l; i++){
    	for(int mask = (1 << l) - 1; mask >= 0;mask--){
    		if(!(mask >> i & 1)){
    			dp1[mask] += dp1[mask + (1 << i)];
    		}
    	}
    }
    for(int i = 0; i < l; i++){
    	for(int mask = 0; mask < (1 << l) ;mask++){
    		if((mask >> i & 1)){
    			dp[mask] += dp[mask - (1 << i)];
    		}
    	}
    }                 
    while(q--){
        string t;                                 
        cin >> t;
        int cnt = 0,cnt1 = 0;
        for(int i = 0; i < l; i++){
        	if(t[i] == '?') cnt++;
        	else if(t[i] == '1') cnt1++;
        }
        if(cnt <= 10){
            int sum = 0,p = 0;
            ll ans = 0;
        	vector <int> v;
        	for(int i = t.size() - 1; i >= 0;i--){
        		if(t[i] == '1'){
        			sum += (1 << p);
        		}
        		else if(t[i] == '?'){
        			v.pb(p);
        		}
        		p++;
        	}
        	for(int mask = 0; mask < (1 << cnt); mask++){
        		int sum1 = 0;
        		for(int j = 0;j < cnt; j++){
        			if((mask >> j & 1) == 1){
        				sum1 += (1 << v[j]);
        			}
        		}
        		ans += s[sum + sum1] - '0';
        	}
        	cout << ans << '\n';
        }               
        else if(cnt1 > 6){
        	int p = 0,sum = 0;
        	ll ans = 0;
        	for(int i =  t.size() - 1; i >= 0; i--){
        		if(t[i] == '1') sum += (1 << p);
        		p++;
        	}
        	ans = dp1[sum];
        	vector <int> v;
        	p = 0;
        	for(int i =  t.size() - 1; i >= 0; i--){
        		if(t[i] == '0') v.pb(p);
        		p++;
        	}
        	for(int mask2 = 1; mask2 < (1 << v.size()); mask2++){
        		int sum1 = sum;
        		for(int j = 0; j < v.size(); j++){
        			if((mask2 >> j & 1) == 1)sum1 += (1 << v[j]);
        		}
        		if(__builtin_popcount(mask2) & 1){
        			ans += dp1[sum1];
        		}
        		else ans -= dp1[sum1];
       		}
       		cout << ans << '\n';
        }  
        else{
        	int p = 0,sum = 0;
        	ll ans = 0;
        	for(int i =  t.size() - 1; i >= 0; i--){
        		if(t[i] == '0') sum += (1 << p);
        		p++;
        	}
        	ans = dp[sum];
        	vector <int> v;
        	p = 0;
        	for(int i =  t.size() - 1; i >= 0; i--){
        		if(t[i] == '1') v.pb(p);
        		p++;
        	}
        	for(int mask2 = 1; mask2 < (1 << v.size()); mask2++){
        		int sum1 = sum;
        		for(int j = 0; j < v.size(); j++){
        			if((mask2 >> j & 1) == 1)sum1 += (1 << v[j]);
        		}
        		if(__builtin_popcount(mask2) & 1){
        			ans += dp[sum1];
        		}
        		else ans -= dp[sum1];
       		}
       		cout << ans << '\n';

        }

    }
    
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...