Submission #210540

#	Time	Username	Problem	Language	Result	Execution time	Memory
210540		kshitij_sodani	Snake Escaping (JOI18_snake_escaping)	C++17	22 / 100	692 ms	40184 KiB

snake_escaping

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
typedef long long llo;
#define a first
#define  b second
#define endl "\n"
int aa=7;
//vector<int> aac[1000001];

//vector<int> pre[16000];
int sl3[21];
int sl[22];
int l;
int dp[1600000];//3^13
int last[1600000];//3^13
int ans[1000001];//q
int it[1100001];//n
int num[1000001];//q
int ok;
/*void find(int ind,int tot,int 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(int ind,int ind2,int 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]);
	}
}
int find3(int ind,int ind3=0){//dp
	if(dp[ind]>-1){
		return dp[ind];
	}
		//cout<<ind<<" ";
	if(last[ind]==-1){
		dp[ind]=it[ind3];
		return dp[ind];
	}
	dp[ind]=find3(ind-sl3[l-aa-last[ind]-1],ind3+sl[l-aa-last[ind]-1])+find3(ind-2*sl3[l-aa-last[ind]-1],ind3);
	return dp[ind];
}
//void(int)
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(int i=1;i<22;i++){
		sl[i]=2*sl[i-1];
	}
	sl3[0]=1;
	for(int i=1;i<14;i++){
		sl3[i]=3*sl3[i-1];
	}

	int q;
	cin>>l>>q;
	aa=0;
	int n=sl[l];
	
	char ss;
	for(int i=0;i<n;i++){
		cin>>ss;
		it[i]=ss-'0';
	}
	

	for(int ii=0;ii<q;ii++){
		for(int 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(int i=0;i<q;i++){
		cout<<num[i]<<" ";
	}
	cout<<endl;*/
	if(l==1){
		return 0;
	}
	/*for(int i=0;i<q;i++){
		find((int)0,(int)0,i);
	}*/
	memset(ans,(int)0,sizeof(ans));
	for(int i=0;i<l-aa;i++){
		find2(0,i,0);
	}

	memset(dp,-1,sizeof(dp));
	//find3(8);
	for(int j=1600000-1;j>=0;j--){
		
		int jj=j;
		int ind5=0;
		int sp=0;
		while(jj){
			if(jj%3==1){
				sp+=sl[ind5];
			}
			jj/=3;
			ind5+=1;
		}

find3(j,sp);
		/*if(j==10){
			cout<<dp[8]<<endl;
		}*/
	}
	/*for(int i=0;i<1;i++){
		ok=i;
		if(pre[i].size()){

			memset(dp,(int)-1,sizeof(dp));
			for(int j=0;j<1600000;j++){
				find3(j);
			}
			for(auto j:pre[i]){
				ans[j]+=it[i];
				//ans[j]+=dp[num[j]];
			//	cout<<i<<":"<<j<<":"<<dp[num[j]]<<endl;

			}
		}

	}*/
	for(int i=0;i<q;i++){
		int jj=num[i];
		int ind5=0;
		int sp=0;
		while(jj){
			if(jj%3==1){
				sp+=sl[ind5];
			}
			jj/=3;
			ind5+=1;
		}


		cout<<find3(num[i],sp)<<endl;
	}









	return 0;
}

• Subtask 1: 5 / 5
• Subtask 2: 7 / 7
• Subtask 3: 10 / 10
• Subtask 4: 0 / 53
• Subtask 5: 0 / 25