Submission #936913

# Submission time Handle Problem Language Result Execution time Memory
936913 2024-03-03T02:46:21 Z vjudge1 Toxic Gene (NOI23_toxic) C++17
68.95 / 100
8 ms 540 KB
#include "toxic.h"
#include <bits/stdc++.h>
#define FOR(i, a, b) for (int i = (a); i < (b); i++)
#define REP(i, n) FOR(i, 0, n)
#define REP1(i, n) FOR(i, 1, n+1)
#define pb push_back
#define ALL(x) (x).begin(), (x).end()
#define SZ(x) (int)((x).size())
using namespace std;
 
/*int query_sample(vector<int> &a){
    cout<<"? ";
    REP(i, SZ(a)) cout<<a[i]<<' ';
    cout<<endl<<flush;
    int in; cin>>in;
    return in;
}
void answer_type(int a, char c){
    cout<<a<<' '<<c<<endl<<flush;
}
*/
 
 
void determine_type(int n){
    mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
 
    int ret;
    vector<int> ids;
    REP1(i, n) ids.pb(i);
    shuffle(ALL(ids), rng);
    vector<int> T, N, S;
 
    vector<int> qq;
    vector<int> dai; // match with T at the very end
 
    int tid = -1, cnt = 0;
    while(SZ(ids)){
        if (cnt == 30){
            while(SZ(ids)){
                dai.pb(ids.back());
                ids.pop_back();
            }
            break;
        }
        //shuffle(ALL(ids), rng);
 
        vector<int> tmp;
        while(SZ(tmp) < 8 && SZ(ids) > 0){
            tmp.pb(ids.back());
            ids.pop_back();
        }
 
        REP(i, SZ(tmp)){
            REP(j, (1<<i)){
                qq.pb(tmp[i]);
            }
        }
        ret = query_sample(qq);
        qq.clear();
 
        if (ret == (1<<SZ(tmp))-1){
            REP(j, SZ(tmp)) dai.pb(tmp[j]);
            continue;
        }
 
        vector<int> tmp2;
        REP(i, SZ(tmp)){
            if ((1<<i)&ret){
                answer_type(tmp[i], 'S');
            }
            else tmp2.pb(tmp[i]);
        }
        // use tmp2 from now on
        if (SZ(tmp2) == 1){
            tid = tmp2[0];
            answer_type(tmp2[0], 'T');
            cnt++;
            continue;
        }
 
        int l = 0, r = SZ(tmp2)-2, mid;
        while(l <= r){
            mid = l+r>>1;
 
            REP(i, mid+1) qq.pb(tmp2[i]);
            ret = query_sample(qq);
            qq.clear();
 
            if (l == r){
                mid = l+(ret != 0);
                break;
            }
 
            if (ret == 0) r = mid-1;
            else l = mid+1;
        }
 
        REP(i, mid){
            answer_type(tmp2[i], 'R');
        }
        answer_type(tmp2[mid], 'T');
        cnt++;
        tid = tmp2[mid];
        FOR(i, mid+1, SZ(tmp2)){
            ids.pb(tmp2[i]);
        }
 
    }
 
    for (int i = 0; i < SZ(dai); i += 8){
        int ci = i;
        vector<int> tmp;
        while(SZ(tmp) < 8 && ci < SZ(dai)){
            REP(j, (1<<(ci-i))){
                qq.pb(dai[ci]);
            }
            tmp.pb(dai[ci]);
            ci++;
        }
        qq.pb(tid);
 
        ret = query_sample(qq);
        qq.clear();
 
        REP(j, SZ(tmp)){
            if ((1<<j)&ret) answer_type(tmp[j], 'S');
            else answer_type(tmp[j], 'R');
        }
    }
 
}
/*
signed main(){
    determine_type(4);
}
*/
// ~270/x * (log2(x)+1) + 30*(log2(x)+1)
 
// ~270/x + (log2(x)+1)*30 - random chances to deduct more from top + remaining ranges need matching

Compilation message

toxic.cpp: In function 'void determine_type(int)':
toxic.cpp:83:20: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   83 |             mid = l+r>>1;
      |                   ~^~
toxic.cpp:81:36: warning: 'mid' may be used uninitialized in this function [-Wmaybe-uninitialized]
   81 |         int l = 0, r = SZ(tmp2)-2, mid;
      |                                    ^~~
# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Partially correct 7 ms 348 KB Partially correct
3 Partially correct 7 ms 348 KB Partially correct
4 Partially correct 6 ms 348 KB Partially correct
5 Partially correct 7 ms 348 KB Partially correct
6 Partially correct 7 ms 348 KB Partially correct
7 Partially correct 7 ms 348 KB Partially correct
8 Partially correct 7 ms 348 KB Partially correct
9 Partially correct 7 ms 540 KB Partially correct
10 Partially correct 7 ms 348 KB Partially correct
11 Correct 6 ms 348 KB Output is correct
12 Partially correct 7 ms 536 KB Partially correct
13 Partially correct 7 ms 348 KB Partially correct
14 Partially correct 7 ms 348 KB Partially correct
15 Partially correct 7 ms 348 KB Partially correct
16 Partially correct 7 ms 348 KB Partially correct
17 Partially correct 7 ms 348 KB Partially correct
18 Partially correct 8 ms 348 KB Partially correct
19 Partially correct 7 ms 348 KB Partially correct
20 Partially correct 7 ms 536 KB Partially correct
21 Correct 6 ms 348 KB Output is correct
22 Correct 6 ms 348 KB Output is correct
23 Correct 5 ms 348 KB Output is correct
24 Partially correct 7 ms 344 KB Partially correct
25 Partially correct 7 ms 348 KB Partially correct
26 Partially correct 7 ms 348 KB Partially correct
27 Partially correct 8 ms 348 KB Partially correct
28 Correct 7 ms 532 KB Output is correct
29 Partially correct 8 ms 536 KB Partially correct
30 Partially correct 7 ms 348 KB Partially correct
31 Partially correct 7 ms 348 KB Partially correct
32 Partially correct 7 ms 348 KB Partially correct
33 Correct 6 ms 536 KB Output is correct
34 Partially correct 7 ms 348 KB Partially correct
35 Partially correct 8 ms 348 KB Partially correct
36 Partially correct 2 ms 440 KB Partially correct