#include "advisor.h"
#include <vector>
void ComputeAdvice(int *C, int N, int K, int M) {
std::vector<int> reuse(K+N,false);
std::vector<int> next(N);
std::vector<int> first(N,N);
for(int i=N-1;i>=0;i--){
next[i]=first[C[i]];
first[C[i]]=i;
}
next.insert(next.begin(),first.begin(),first.begin()+K);
std::vector<int> colors;
for(int i=0;i<K;i++){
colors.push_back(i);
}
colors.insert(colors.end(),C,C+N);
//time entered: [0,K) for init, [K,N+K) for added
//time color will be reused: next[cache[i]]
std::vector<int> cache(K);
for(int i=0;i<K;i++){
cache[i]=i;
}
for(int i=0;i<N;i++){
int late=0;
for(int k=0;k<K;k++){
if(colors[cache[k]]==C[i]){
late=k;
break;
}
if(next[cache[k]]>next[cache[late]]){
late=k;
}
}
if(colors[cache[late]]==C[i]){
reuse[cache[late]]=true;
}
//evict cache[late]
cache[late]=i+K;
}
for(int i=0;i<K+N;i++){
WriteAdvice(reuse[i]);
}
}
#include "assistant.h"
#include <vector>
void Assist(unsigned char *A, int N, int K, int R) {
std::vector<int> advice(A,A+R);
std::vector<std::pair<int,int> > cache(K);//(color,reused)
for(int i=0;i<K;i++){
cache[i]={i,advice[i]};
}
for(int i=0;i<N;i++){
int color=GetRequest();
int evict=0;
for(int k=0;k<K;k++){
if(cache[k].first==color){
evict=k;
break;
}
if(!cache[k].second){
evict=k;
}
}
if(cache[evict].first!=color){
PutBack(cache[evict].first);
}
cache[evict].first=color;
cache[evict].second=advice[i+K];
}
}
