| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 210500 | kshitij_sodani | Snake Escaping (JOI18_snake_escaping) | C++17 | 0 ms | 0 KiB |
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 <iostream>
#include <bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
typedef long long int llo;
#define a first
#define b second
#define endl "\n"
llo aa=7;
vector<llo> aac[1000001];
vector<llo> pre[130];
llo sl3[21];
llo sl[22];
llo l;
llo dp[1600000];
llo last[1600000];
llo ans[1000000];
llo it[1200000];
llo num[1200000];
llo ok;
void find(llo ind,llo tot,llo ind2){//pre_sub
if(ind==aa){
pre[tot].pb(ind2);
// cout<<tot<<" "<<ind2<<endl;
}
else{
if(aac[ind2][ind]%2==0){
find(ind+1,tot,ind2);
}
if(aac[ind2][ind]>0){
find(ind+1,tot+sl[aa-ind-1],ind2);
}
}
}
void find2(llo ind,llo ind2,llo tot=0){//
if(ind==l-aa){
last[tot]=ind2;
// cout<<tot<<","<<ind2<<endl;
return;
}
if(ind==ind2){
find2(ind+1,ind2,tot+2*sl3[(l-aa)-ind-1]);
}
else{
find2(ind+1,ind2,tot);
find2(ind+1,ind2,tot+sl3[(l-aa)-ind-1]);
}
if(ind<ind2){
find2(ind+1,ind2,tot+2*sl3[(l-aa)-ind-1]);
}
}
llo find3(llo ind){//dp
if(dp[ind]>-1){
return dp[ind];
}
if(last[ind]==-1){
dp[ind]=it[ind+ok*(sl[l-aa])];
return dp[ind];
}
dp[ind]=find3(ind-sl3[l-aa-last[ind]-1])+find3(ind-2*sl3[l-aa-last[ind]-1]);
return dp[ind];
}
//void(llo)
int main(){
ios_base::sync_with_stdio(false);
cin.tie(NULL);
memset(last,-1,sizeof(last));
memset(num,0,sizeof(num));
sl[0]=1;
for(llo i=1;i<22;i++){
sl[i]=2*sl[i-1];
}
sl3[0]=1;
for(llo i=1;i<14;i++){
sl3[i]=3*sl3[i-1];
}
llo q;
cin>>l>>q;
aa=min(7,l-1);
llo n=1<<l;
char ss;
for(llo i=0;i<n;i++){
cin>>ss;
it[i]=ss-'0';
}
for(llo ii=0;ii<q;ii++){
for(llo i=0;i<l;i++){
cin>>ss;
if(ss=='0'){
aac[ii].pb(0);
if(l==1){
cout<<it[0]<<endl;
}
}
else if(ss=='1'){
aac[ii].pb(1);
if(l==1){
cout<<it[1]<<endl;
}
else{
if(i>=aa){
num[ii]+=sl3[l-i-1];
}
}
}
else{
aac[ii].pb(2);
if(l==1){
cout<<it[0]+it[1]<<endl;
}
else{
if(i>=aa){
num[ii]+=2*sl3[l-i-1];
}
}
}
}
}
/*for(llo i=0;i<q;i++){
cout<<num[i]<<" ";
}
cout<<endl;*/
if(l==1){
return 0;
}
for(llo i=0;i<q;i++){
find(0,0,i);
}
memset(ans,0,sizeof(ans));
for(llo i=0;i<l-aa;i++){
find2(0,i,0);
}
for(llo i=0;i<130;i++){
if(pre[i].size()==0){
continue;
}
}
for(llo i=0;i<130;i++){
ok=i;
if(pre[i].size()){
memset(dp,-1,sizeof(dp));
for(llo j=0;j<1600000;j++){
find3(j);
}
for(auto j:pre[i]){
ans[j]+=dp[num[j]];
// cout<<i<<":"<<j<<":"<<dp[num[j]]<<endl;
}
}
}
for(llo i=0;i<q;i++){
cout<<ans[i]<<endl;
}
return 0;
}