Submission #941038

#TimeUsernameProblemLanguageResultExecution timeMemory
941038socpiteToxic Gene (NOI23_toxic)C++17
51.40 / 100
10 ms660 KiB
#include<bits/stdc++.h>
#include"toxic.h"
using namespace std;

mt19937 rng(69420);

int query_sample(vector<int> species);

int h[301];

int shuffled_query(vector<int> speices){
	for(auto &v: speices)v = h[v];
	return query_sample(speices);
}

void shuffled_answer(int x, char c){
	answer_type(h[x], c);
}

void determine_type(int n){
    int l = 1, r = n;
	iota(h+1, h+n+1, 1);
	// shuffle(h+1, h+n+1, rng);
    vector<int> vec(n);
	vector<bool> bad(n+1, 0);
    vector<bool> done(n+1, 0);
    iota(vec.begin(), vec.end(), 1);
    while(l < r){
        int mid = (l+r)>>1;
		int tmp = shuffled_query(vector<int>(vec.begin() + l - 1, vec.begin() + mid));
        if(tmp != mid - l + 1)r = mid;
        else l = mid+1;
    }
    vec.erase(vec.begin() + l - 1);
    shuffled_answer(l, 'T');
	// vector<int> missing;
	int tcnt = 0;
    while(!vec.empty()){
		// shuffle(vec.begin(), vec.end(), rng);
        vector<int> sampling;
        vector<int> species;
        vector<int> missing;
        for(int i = 0; i < min<int>(8, vec.size()); i++){
            sampling.push_back(vec[i]);
            for(int _ = 0; _ < (1<<i); _++)species.push_back(vec[i]);
        }
		for(int i = 0; i < sampling.size(); i++)vec.erase(vec.begin());
		bool all_bad = 1;
		for(auto v: sampling)if(!bad[v])all_bad = 0;
		if(all_bad && tcnt == 30){
			for(auto v: sampling)shuffled_answer(v, 'R');	
		}
        int tot1 = shuffled_query(species);
        if(tot1 == species.size()){
            species.push_back(l);
			if(all_bad || tcnt == 30){
				for(auto v: sampling)shuffled_answer(v, 'R');
				continue;
			}
            int tot2 = shuffled_query(species);
            for(int i = 0; i < sampling.size(); i++){
                if((tot1&tot2)&(1<<i))shuffled_answer(sampling[i], 'S');
                else shuffled_answer(sampling[i], 'R');
            }
        }
        else {
            for(int i = 0; i < sampling.size(); i++){
                if(tot1&(1<<i)){
                    shuffled_answer(sampling[i], 'S');
                }
                else missing.push_back(sampling[i]);
            }
            int ql = 0, r = missing.size()-1;
            while(ql < r){
                int mid = (ql+r)>>1;
                if(!shuffled_query(vector<int>(missing.begin(), missing.begin() + mid + 1)))r = mid;
                else ql = mid+1;
            }
			tcnt++;
            for(int i = 0; i <= ql; i++){
                if(i < ql)shuffled_answer(missing[i], 'R');
                else shuffled_answer(missing[i], 'T');
            }
			for(int i = ql+1; i < missing.size(); i++){
				bad[missing[i]] = 1;
				vec.push_back(missing[i]);
			}
        }
    }
}

Compilation message (stderr)

toxic.cpp: In function 'void determine_type(int)':
toxic.cpp:47:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   47 |   for(int i = 0; i < sampling.size(); i++)vec.erase(vec.begin());
      |                  ~~^~~~~~~~~~~~~~~~~
toxic.cpp:54:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   54 |         if(tot1 == species.size()){
      |            ~~~~~^~~~~~~~~~~~~~~~~
toxic.cpp:61:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   61 |             for(int i = 0; i < sampling.size(); i++){
      |                            ~~^~~~~~~~~~~~~~~~~
toxic.cpp:67:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   67 |             for(int i = 0; i < sampling.size(); i++){
      |                            ~~^~~~~~~~~~~~~~~~~
toxic.cpp:84:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   84 |    for(int i = ql+1; i < missing.size(); i++){
      |                      ~~^~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...