# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
871695 | 2023-11-11T10:33:48 Z | Matjaz | Last supper (IOI12_supper) | C++14 | 0 ms | 0 KB |
#include "advisor.h" #include<set> #include<queue> #include<vector> using namespace std; void ComputeAdvice(int *C, int N, int K, int M) { set<int> scafold; for (int i=0;i<K;i++) scafold.insert(i); vector<queue<int> > Q(N); vector<queue<int> > returns(N); for (int i=0;i<N;i++) Q[C[i]].push(i); for (int i=0;i<N;i++) Q[i].push(N); vector<int> takeback(N, -1); for (int i=0;i<N;i++){ Q[C[i]].pop(); if (scafold.count(C[i]) > 0) continue; int besttime = -1; int bestcolour = -1; for (set<int>::iterator it = scafold.begin(); it != scafold.end(); it++){ int u = *it; if (Q[u].empty() || Q[u].front() > besttime){ besttime = Q[u].empty() ? N : Q[u].front(); bestcolour = u; } } scafold.erase(bestcolour); scafold.insert(C[i]); returns[bestcolour].push(i); takeback[i] = bestcolour; } for (int i=0;i<N;i++) Q[i].clear(); for (int i=0;i<N;i++) Q[C[i]].push(i); for (int i=0;i<N;i++) Q[i].push(N); for (int i=0;i<K;i++){ if (Q[i].empty()){ WriteAdvice(0); continue; } if (returns[i].empty()){ WriteAdvice(1); continue; } if (Q[i].front() <= returns[i].front()){ WriteAdvice(1); } else WriteAdvice(0); } for (int i=0;i<N;i++){ Q[C[i]].pop(); if (Q[C[i]].empty()){ WriteAdvice(0); continue; } if (returns[C[i]].empty()){ WriteAdvice(1); continue; } if (Q[C[i]].front() <= returns[C[i]].front()){ WriteAdvice(1); } else WriteAdvice(0); returns[takeback[i]].pop(); } }