oj.uz

Submission #565910

# Submission time Handle Problem Language Result Execution time Memory
565910 2022-05-21T14:13:12 Z Deepesson ICC (CEOI16_icc) C++17
100 / 100
 122 ms 520 KB 
#include <bits/stdc++.h>

#include "icc.h"

#define MAX 105

typedef std::pair<int,int> pii;

int query(int size_a, int size_b, int a[], int b[]);

int query(std::vector<int> a,std::vector<int> b){
	for(auto&x:a)++x;
	for(auto&x:b)++x;
	return query(a.size(),b.size(),a.data(),b.data());
}

void setRoad(int a, int b);
void InformaRua(int a,int b){setRoad(a+1,b+1);}
int pai[MAX];

int find(int a){
	if(pai[a]==a){
		return a;
	}
	return pai[a]=find(pai[a]);
}

void Union(int a,int b){
	a=find(a);
	b=find(b);
	pai[a]=b;
}

std::vector<std::vector<int>> gerar_grupos(int N){
	std::map<int,std::vector<int>> mapa;
	for(int i=0;i!=N;++i){
		mapa[find(i)].push_back(i);
	}
	std::vector<std::vector<int>> res;
	for(auto&x:mapa){
		res.push_back(x.second);
	}
	return res;
}

int splits[20];
int lvl;
int andar[MAX];
void dnc(int l,int r,int nv){
	if(l==r)return;
	lvl=std::max(lvl,nv);
	int m = (l+r)/2;
	for(int i=l;i<=m;++i)andar[i]+=(1<<nv);
	dnc(l,m,nv+1);
	dnc(m+1,r,nv+1);
}

void add(std::vector<int>& ref,std::vector<int> cp){
	for(auto&x:cp)ref.push_back(x);
}

void pruning(std::vector<int>& a,std::vector<int> b){
	while(a.size()!=1){
		int m = a.size()/2;
		std::vector<int> c,d;
		for(int i=0;i<m;++i){
			c.push_back(a[i]);
		}
		for(int i=m;i!=a.size();++i){
			d.push_back(a[i]);
		}
		if(query(c,b)){
			a=c;
		}else a=d;
	}
}
void run(int N) {
	for(int i=0;i!=MAX;++i)pai[i]=i;

	for(int i=1;i!=N;++i){
		auto grups = gerar_grupos(N);
		memset(andar,0,sizeof(andar));
		lvl=0;
		dnc(0,grups.size()-1,0);
		std::random_shuffle(grups.begin(),grups.end());
		std::vector<int> l,r;
		for(int j=0;j!=20;++j){
			std::vector<int> a,b;
			for(int k=0;k!=grups.size();++k){
				if(!(andar[k]&(1<<j))){
					add(a,grups[k]);
				}else add(b,grups[k]);
			}
			if(j==lvl||query(a,b)){
				l=a;
				r=b;
				goto prox;
			}
		}
		prox:
		pruning(l,r);
		pruning(r,l);
		InformaRua(l[0],r[0]);
		Union(l[0],r[0]);
	}
}

Compilation message

icc.cpp: In function 'void pruning(std::vector<int>&, std::vector<int>)':
icc.cpp:69:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   69 |   for(int i=m;i!=a.size();++i){
      |               ~^~~~~~~~~~
icc.cpp: In function 'void run(int)':
icc.cpp:89:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   89 |    for(int k=0;k!=grups.size();++k){
      |                ~^~~~~~~~~~~~~~

Subtask #1
7.0 / 7.0

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

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 28 ms 468 KB Ok! 526 queries used.
2 Correct 32 ms 468 KB Ok! 635 queries used.
3 Correct 32 ms 468 KB Ok! 642 queries used.

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 108 ms 500 KB Ok! 1458 queries used.
2 Correct 119 ms 492 KB Ok! 1587 queries used.
3 Correct 113 ms 516 KB Ok! 1564 queries used.
4 Correct 107 ms 492 KB Ok! 1526 queries used.

Subtask #4
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 104 ms 500 KB Ok! 1494 queries used.
2 Correct 106 ms 520 KB Ok! 1482 queries used.
3 Correct 115 ms 500 KB Ok! 1571 queries used.
4 Correct 108 ms 516 KB Ok! 1415 queries used.

Subtask #5
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 110 ms 512 KB Ok! 1578 queries used.
2 Correct 116 ms 468 KB Ok! 1612 queries used.
3 Correct 122 ms 500 KB Ok! 1606 queries used.
4 Correct 113 ms 520 KB Ok! 1591 queries used.
5 Correct 110 ms 468 KB Ok! 1525 queries used.
6 Correct 115 ms 500 KB Ok! 1607 queries used.

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 111 ms 496 KB Ok! 1584 queries used.
2 Correct 114 ms 496 KB Ok! 1570 queries used.