# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1024076 | MrAndria | CONSUL (info1cup19_consul) | C++14 | 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 <bits/stdc++.h>
using namespace std;
#define pb push_back
static const int MIN_VALUE = 0, MAX_VALUE = (1e9) - 1;
static map<int,int> mp;
static int Q, N, a[5005];
static bool issol, answer;
vector <int> v;
void say_answer(int k)
{
if(answer)
{
cout << "Multiple answers provided for the same testcase!\n";
exit(0);
}
answer = 1;
if(k == -1)
{
if(issol)
{
cout << "Wrong answer\n";
exit(0);
}
else cout << "Correct! Number of queries: " << Q << '\n';
}
else
{
if(!issol || mp[k] <= N/3)
{
cout << "Wrong answer\n";
exit(0);
}
else cout << "Correct! Number of queries: " << Q << '\n';
}
}
int cnt(int k)
{
++Q;
if(!(k>=MIN_VALUE && k<=MAX_VALUE))
{
cout << "Wrong query format\n";
exit(0);
}
return mp[k];
}
int kth(int k)
{
++Q;
if(!(k>=1 && k<=N))
{
cout << "Wrong query format\n";
exit(0);
}
return a[k];
}
void solve(int N){
srand(time(0));
for(int i=1;i<=N;i++){
v.pb(i);
}
int q=30;
while(v.size() and q--){
int k,l,num;
k=rand();
k%=(v.size());
l=kth(v[k]);
num=cnt(l);
if(num>N/3){
say_answer(l);
return;
}
v.erase(v.begin()+k);
}
say_answer(-1);
return;
}
int main()
{
int tests, i;
cin >> tests;
while(tests--)
{
cin >> N; mp.clear();
Q = 0; issol = 0; answer = 0;
for(i=1; i<=N; ++i) cin >> a[i], ++mp[a[i]];
for(i=1; i<=N; ++i) issol |= (mp[a[i]] > N/3);
solve(N);
}
return 0;
}