oj.uz

Submission #605638

# Submission time Handle Problem Language Result Execution time Memory
605638 2022-07-25T20:41:59 Z AlperenT ICC (CEOI16_icc) C++17
90 / 100
 142 ms 496 KB 
#include <bits/stdc++.h>
#include "icc.h"

using namespace std;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

struct Comp{
	vector<int> nodes;
};

vector<vector<int>> tree;

vector<bool> vis;

void dfs(int v, vector<int> &vec){
	vis[v] = true;
	vec.push_back(v);

	for(auto e : tree[v]){
		if(!vis[e]) dfs(e, vec);
	}
}

bool cmp(Comp &a, Comp &b){
	return a.nodes.size() < b.nodes.size();
}

void run(int n){
	tree.assign(n + 1, {});

	int mx = 0;

	for(int it = 0; it <= n - 1; it++){
		int s, t;

		vector<Comp> comps;

		vis.assign(n + 1, false);

		for(int v = 1; v <= n; v++){
			if(!vis[v]){
				Comp tmp;

				comps.push_back(tmp);
				dfs(v, comps.back().nodes);
			}
		}

		sort(comps.begin(), comps.end(), cmp);

		vector<int> a, b, c;

		for(int k = 0; k < 7; k++){
			a.clear(), b.clear();

			for(int i = 0; i < comps.size(); i++){
				if(i & (1 << k)) for(auto v : comps[i].nodes) a.push_back(v);
				else for(auto v : comps[i].nodes) b.push_back(v);
			}

			if(query(a.size(), b.size(), a.data(), b.data())) break;
		}

		shuffle(a.begin(), a.end(), rng);
		shuffle(b.begin(), b.end(), rng);

		int l = -1, r = a.size() - 1, m;

		while(r - l > 1){
			m = l + (r - l) / 2;

			c.clear();

			for(int i = 0; i <= m; i++) c.push_back(a[i]);

			if(query(c.size(), b.size(), c.data(), b.data())) r = m;
			else l = m;
		}

		s = a[r];

		l = -1, r = b.size() - 1;

		while(r - l > 1){
			m = l + (r - l) / 2;

			c.clear();

			for(int i = 0; i <= m; i++) c.push_back(b[i]);

			if(query(c.size(), a.size(), c.data(), a.data())) r = m;
			else l = m;
		}

		t = b[r];

		setRoad(s, t);

		tree[s].push_back(t);
		tree[t].push_back(s);
	}
}

Compilation message

icc.cpp: In function 'void run(int)':
icc.cpp:57:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<Comp>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |    for(int i = 0; i < comps.size(); i++){
      |                   ~~^~~~~~~~~~~~~~
icc.cpp:32:6: warning: unused variable 'mx' [-Wunused-variable]
   32 |  int mx = 0;
      |      ^~

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 468 KB Ok! 100 queries used.
2 Correct 4 ms 468 KB Ok! 97 queries used.

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 26 ms 468 KB Ok! 519 queries used.
2 Correct 37 ms 468 KB Ok! 707 queries used.
3 Correct 40 ms 468 KB Ok! 698 queries used.

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 104 ms 488 KB Ok! 1381 queries used.
2 Correct 128 ms 492 KB Ok! 1742 queries used.
3 Correct 115 ms 492 KB Ok! 1469 queries used.
4 Correct 129 ms 488 KB Ok! 1509 queries used.

Subtask #4
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 116 ms 468 KB Ok! 1488 queries used.
2 Correct 119 ms 496 KB Ok! 1520 queries used.
3 Correct 129 ms 492 KB Ok! 1705 queries used.
4 Correct 126 ms 468 KB Ok! 1440 queries used.

Subtask #5
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 142 ms 488 KB Ok! 1711 queries used.
2 Correct 130 ms 496 KB Ok! 1721 queries used.
3 Correct 133 ms 488 KB Ok! 1714 queries used.
4 Correct 127 ms 496 KB Ok! 1680 queries used.
5 Correct 126 ms 488 KB Ok! 1485 queries used.
6 Correct 124 ms 488 KB Ok! 1543 queries used.

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 135 ms 468 KB Too many queries! 1718 out of 1625
2 Halted 0 ms 0 KB -