Submission #116711

# Submission time Handle Problem Language Result Execution time Memory
116711 2019-06-13T15:47:29 Z claudy Library (JOI18_library) C++14
100 / 100
537 ms 632 KB
# pragma GCC optimize("Ofast")
# pragma GCC optimization ("unroll-loops")
# include "bits/stdc++.h"
std::pair<int,int> DR[] = {{-1,0},{0,1},{1,0},{0,-1},{-1,1},{-1,-1},{1,1},{1,-1}};
# define ll long long
# define clock (clock() * 1000.0 / CLOCKS_PER_SEC)
# define rc(s) return cout << s,0
# define rcg(s) cout << s;exit(0)
# define _ ios_base::sync_with_stdio(false);cin.tie(0);cerr.tie(0);cout.tie(0);
# define db(x) cerr << #x << " = " << x << '\n'
# define pb push_back
# define mp make_pair
# define all(s) s.begin(),s.end()
# define sz(x) (int)((x).size())
//# define int ll
using namespace std;

// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
# define LOCAL
# define sim template < class c
# define ris return * this
# define dor > debug & operator <<
# define eni(x) sim > typename \
enable_if<sizeof dud<c>(0) x 1, debug&>::type operator<<(c i) {
sim > struct rge { c b, e; };
sim > rge<c> range(c i, c j) { return rge<c>{i, j}; }
sim > auto dud(c* x) -> decltype(cerr << *x, 0);
sim > char dud(...);
struct debug {
#ifdef LOCAL
~debug() { cerr << endl; }
eni(!=) cerr << boolalpha << i; ris; }
eni(==) ris << range(begin(i), end(i)); }
sim, class b dor(pair < b, c > d) {
ris << "(" << d.first << ", " << d.second << ")";
}
sim dor(rge<c> d) {
*this << "[";
for (auto it = d.b; it != d.e; ++it)
*this << ", " + 2 * (it == d.b) << *it;
ris << "]";
}
#else
sim dor(const c&) { ris; }
#endif
};
#define show(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] "
int gcd(int a, int b)
{
if(b) return gcd(b,a%b);
return a;
}mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

int Query(const std::vector<int>& M);
void Answer(const std::vector<int>& res);
# include "library.h"

/*int Query(vector<int>vec)
{
	for(int i = 0;i < sz(vec);i++)
	{
		if(vec[i]) cout << i + 1 << ' ';
	}
	cout << endl;
	int x;
	cin >> x;
	return x;
}

void Answer(vector<int>vec)
{
	for(auto it : vec) cout << it << ' ';
}*/

set<int>s;
vector<int>M;
int n;
vector<int>vec;
int k = 0;
bool B = 0;

void solve(int x,int par)
{
	if(x == -1) return;
	s.erase(x);
	vec.pb(x);
	int l = 0;
	int r = sz(s) - 1;
	while(l < r)
	{
		int mid = l + r >> 1;
		for(int i = 0;i < n;i++)
		{
			M[i] = 0;
		}
		M[par - 1] = 1;
		auto it = s.begin();
		for(int i = 0;i <= mid;i++)
		{
			M[*it - 1] = 1;
			it++;
		}
		int a = Query(M);
		M[x - 1] = 1;
		int b = Query(M);
		if(a != b) r = mid;
		else l = mid + 1;
	}
	if(l > r) {
		solve(-1,x);
		return ;
	}
	auto it = s.begin();
	if(!B){
	int mid = l;
	for(int i = 0;i < n;i++)
	{
		M[i] = 0;
	}
	M[par - 1] = 1;
	for(int i = 0;i <= mid;i++)
	{
		M[*it - 1] = 1;
		it++;
	}
	int a = Query(M);
	M[x - 1] = 1;
	int b = Query(M);
	if(a == b)
	{
		solve(-1,x);
		B = 1;
		return;
	}
}
	it = s.begin();
	for(int i = 0;i <= l;i++)
	{
		if(i == l) {
			solve(*it,x);
			return ;
		}
		it++;
	}
}

void Solve(int N)
{
	n = N;
	if(n == 1)
	{
		vector<int>vec;vec.pb(1);Answer(vec);return;
	}
	M.resize(n);
	int x = 1 + (rng() % n);
	M[x - 1] = 1;
	vector<int>adj;
	for(int i = 1;i <= n;i++)
	{
		if(sz(adj) == 2) break;
		if(i != x){
		M[i - 1] = 1;
		if(Query(M) == 1) adj.pb(i);
		M[i - 1] = 0;
		}
	}
	for(int i = 1;i <= n;i++)
	{
		if(i != x) s.insert(i);
	}
	vec.pb(x);
	solve(adj[0],x);
	vector<int>ans;
	for(auto it : vec) ans.pb(it);
	vec.clear();
	if(sz(adj) > 1)
	{
		solve(adj[1],x);
		reverse(vec.begin(),vec.end());
		for(auto it : ans) vec.pb(it);
		Answer(vec);
	}
	else Answer(ans);
}





Compilation message

library.cpp:2:0: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
 # pragma GCC optimization ("unroll-loops")
 
library.cpp: In function 'void solve(int, int)':
library.cpp:92:15: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   int mid = l + r >> 1;
             ~~^~~
# Verdict Execution time Memory Grader output
1 Correct 48 ms 256 KB # of queries: 2459
2 Correct 43 ms 384 KB # of queries: 2492
3 Correct 41 ms 376 KB # of queries: 2734
4 Correct 34 ms 412 KB # of queries: 2879
5 Correct 36 ms 256 KB # of queries: 2997
6 Correct 41 ms 256 KB # of queries: 2776
7 Correct 27 ms 336 KB # of queries: 2780
8 Correct 27 ms 328 KB # of queries: 2730
9 Correct 53 ms 444 KB # of queries: 2978
10 Correct 28 ms 384 KB # of queries: 1577
11 Correct 2 ms 256 KB # of queries: 0
12 Correct 2 ms 256 KB # of queries: 1
13 Correct 2 ms 384 KB # of queries: 4
14 Correct 2 ms 256 KB # of queries: 6
15 Correct 3 ms 284 KB # of queries: 79
16 Correct 4 ms 384 KB # of queries: 248
# Verdict Execution time Memory Grader output
1 Correct 48 ms 256 KB # of queries: 2459
2 Correct 43 ms 384 KB # of queries: 2492
3 Correct 41 ms 376 KB # of queries: 2734
4 Correct 34 ms 412 KB # of queries: 2879
5 Correct 36 ms 256 KB # of queries: 2997
6 Correct 41 ms 256 KB # of queries: 2776
7 Correct 27 ms 336 KB # of queries: 2780
8 Correct 27 ms 328 KB # of queries: 2730
9 Correct 53 ms 444 KB # of queries: 2978
10 Correct 28 ms 384 KB # of queries: 1577
11 Correct 2 ms 256 KB # of queries: 0
12 Correct 2 ms 256 KB # of queries: 1
13 Correct 2 ms 384 KB # of queries: 4
14 Correct 2 ms 256 KB # of queries: 6
15 Correct 3 ms 284 KB # of queries: 79
16 Correct 4 ms 384 KB # of queries: 248
17 Correct 492 ms 428 KB # of queries: 18238
18 Correct 485 ms 548 KB # of queries: 19294
19 Correct 537 ms 576 KB # of queries: 18960
20 Correct 381 ms 580 KB # of queries: 17358
21 Correct 386 ms 560 KB # of queries: 16089
22 Correct 495 ms 632 KB # of queries: 19552
23 Correct 450 ms 576 KB # of queries: 18457
24 Correct 133 ms 384 KB # of queries: 7921
25 Correct 448 ms 528 KB # of queries: 17701
26 Correct 484 ms 632 KB # of queries: 17841
27 Correct 181 ms 388 KB # of queries: 8907
28 Correct 459 ms 500 KB # of queries: 19235
29 Correct 462 ms 436 KB # of queries: 19583
30 Correct 527 ms 504 KB # of queries: 19869