Submission #575948

#	Time	Username	Problem	Language	Result	Execution time	Memory
575948		PedroBigMan	Crayfish scrivener (IOI12_scrivener)	C++14	34 / 100	1090 ms	167056 KiB

scrivener

/*
Author of all code: Pedro BIGMAN Dias
Last edit: 15/02/2021
*/
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#pragma GCC optimize("Ofast")
#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
#include <string>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <queue>
#include <deque>
#include <list>
#include <iomanip>
#include <stdlib.h>
#include <time.h>
#include <cstring>
using namespace std;
typedef int ll;
typedef unsigned long long int ull;
typedef long double ld;
#define REP(i,a,b) for(ll i=(ll) a; i<(ll) b; i++)
#define pb push_back
#define mp make_pair
#define pl pair<ll,ll>
#define ff first
#define ss second
#define whole(x) x.begin(),x.end()
#define DEBUG(i) cout<<"Pedro Is The Master "<<i<<endl
#define INF 1000000000000000000LL
#define EPS ((ld)0.00000000001)
#define pi ((ld)3.141592653589793)
#define VV(vvvv,NNNN,xxxx); REP(iiiii,0,NNNN) {vvvv.pb(xxxx);}
ll mod=1000000007LL;

template<class A=ll> 
void Out(vector<A> a) {REP(i,0,a.size()) {cout<<a[i]<<" ";} cout<<endl;}

template<class A=ll>
void In(vector<A> &a, ll N) {A cur; REP(i,0,N) {cin>>cur; a.pb(cur);}} 

int S[1000000];
pl s[1000000];
pl f[1000000][21]; //f2[i][exp] = {node after 2^exp moves, number of letters}
int C;

void Init() 
{
	C=0;
	return;
}

void AddState(ll val, ll nxt)
{
	s[C]={val,nxt}; 
	if(C==0) {S[C]=1;} else {S[C]=S[nxt]+min(val+1,1);}
	C++;
	ll exp=0; ll curval = min(s[nxt].ff+1,1); if(C==1) {curval=0;} ll ind = C-1;
	f[ind][0]={nxt,curval};
	REP(exp,1,21)
	{
		f[ind][exp]={f[f[ind][exp-1].ff][exp-1].ff,f[ind][exp-1].ss+f[f[ind][exp-1].ff][exp-1].ss};
	}
}

void TypeLetter(char L) 
{
	ll val = (ll) (L-'a');
	AddState(val,max(0,(int)(C-1)));
}

void UndoCommands(int U) 
{
	AddState(-1,C-1-U);
}

char GetLetter(int P) 
{
	ll needed = S[C-1]-P; 
	ll curind=C-1; ll curval = min(1,s[curind].ff+1); 
	ll exp = 20; ll poss; ll nxt;
	//REP(i,0,s.size()) {REP(j,0,f[i].size()) {cout<<f[i][j].ff<<" "<<f[i][j].ss<<endl;} cout<<endl;}
	ll ans=-1; if(curval==needed) {ans=s[curind].ff;}
	while(curval<needed)
	{
		poss=curval+f[curind][exp].ss;
		if(poss<needed) {curval=poss; curind = f[curind][exp].ff; exp--; continue;}
		if(poss>needed) {exp--; continue;}
		nxt = f[curind][exp].ff; if(s[nxt].ff>=0) {ans = s[nxt].ff; break;} exp--;
	}
	if(ans>=0) {return ((char) ans+'a');}
	return ans;
}

Compilation message (stderr)

scrivener.cpp:5: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    5 | #pragma GCC optimization ("O3")
      | 
scrivener.cpp:6: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    6 | #pragma GCC optimization ("unroll-loops")
      | 
scrivener.cpp: In function 'void AddState(ll, ll)':
scrivener.cpp:64:5: warning: unused variable 'exp' [-Wunused-variable]
   64 |  ll exp=0; ll curval = min(s[nxt].ff+1,1); if(C==1) {curval=0;} ll ind = C-1;
      |     ^~~

• Subtask 1: 5 / 5
• Subtask 2: 7 / 7
• Subtask 3: 22 / 22
• Subtask 4: 0 / 26
• Subtask 5: 0 / 40