oj.uz

답안 #124129

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
124129 2019-07-02T14:42:44 Z eriksuenderhauf CEOI16_icc (CEOI16_icc) C++11
100 / 100
 159 ms 660 KB 
//#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#include "icc.h"
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/rope>
#define mem(a,v) memset((a), (v), sizeof (a))
#define enl printf("\n")
#define case(t) printf("Case #%d: ", (t))
#define ni(n) scanf("%d", &(n))
#define nl(n) scanf("%I64d", &(n))
#define nai(a, n) for (int i = 0; i < (n); i++) ni(a[i])
#define nal(a, n) for (int i = 0; i < (n); i++) nl(a[i])
#define pri(n) printf("%d\n", (n))
#define prl(n) printf("%I64d\n", (n))
#define pii pair<int, int>
#define pil pair<int, long long>
#define pll pair<long long, long long>
#define vii vector<pii>
#define vil vector<pil>
#define vll vector<pll>
#define vi vector<int>
#define vl vector<long long>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef cc_hash_table<int,int,hash<int>> ht;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> oset;
const double pi = acos(-1);
const int MOD = 1e9 + 7;
const int INF = 1e9 + 7;
const int MAXN = 1e2 + 5;
const double eps = 1e-9;
int par[MAXN], sz[MAXN];
vi comp[MAXN];

int qry(int x) { return par[x] == x ? par[x] : qry(par[x]); }
void join(int u, int v) {
	u = qry(u), v = qry(v);
	if (u == v) return;
	if (sz[u] > sz[v]) swap(u, v);
	par[u] = v;
	comp[v].insert(comp[v].end(), comp[u].begin(), comp[u].end());
}

void print(vi cur) {
	for (int i: cur) cerr << i << " "; cerr << "\n";
}

pii solve(int N) {
	pii ret;
	vi cand;
	for (int i = 1; i <= N; i++) {
		if (qry(i) != i)
			continue;
		cand.pb(i);
	}
	vi l, r;
	int msk = 0;
	for (int ind = 0; (1<<ind) < (int)cand.size(); ind++) {
		l.clear();
		r.clear();
		for (int j = 0; j < (int)cand.size(); j++) {
			if ((j >> ind) & 1)
				l.insert(l.end(), comp[cand[j]].begin(), comp[cand[j]].end());
			else
				r.insert(r.end(), comp[cand[j]].begin(), comp[cand[j]].end());
		}
		int la[l.size()], ra[r.size()];
		copy(l.begin(), l.end(), la);
		copy(r.begin(), r.end(), ra);
		int fl = query((int)l.size(),(int)r.size(), la, ra);
		msk |= fl * (1 << ind);
	}
	l.clear();
	r.clear();
	for (int i = 0; i < (int)cand.size(); i++) {
		if (i & (msk & -msk))
			l.insert(l.end(), comp[cand[i]].begin(), comp[cand[i]].end());
		else if ((i ^ msk) < (int)cand.size())
			r.insert(r.end(), comp[cand[i]].begin(), comp[cand[i]].end());
	}
	int lo = 0, hi = l.size()-1;
	if (hi == lo)
		hi--;
	while (lo <= hi) {
		int mi = (lo+hi) / 2;
		vi tmp;
		for (int i = 0; i <= mi; i++)
			tmp.pb(l[i]);
		int la[tmp.size()], ra[r.size()];
		copy(tmp.begin(), tmp.end(), la);
		copy(r.begin(), r.end(), ra);
		int fl = query((int)tmp.size(),(int)r.size(), la, ra);
		if (fl == 1)
			hi = mi - 1;
		else
			lo = mi + 1;
	}
	ret.fi = l[lo];
	int ind = 0;
	for (; ind < (int)cand.size(); ind++)
		if (cand[ind] == qry(l[lo]))
			break;
	//cerr << l[lo] << " " << ind << " " << msk << "\n";
	//print(l);
	//print(r);
	if (ind & (msk & -msk)) {
		r.clear();
		r.insert(r.end(), comp[cand[ind^msk]].begin(), comp[cand[ind^msk]].end());
	} else {
		l.clear();
		l.insert(l.end(), comp[cand[ind^msk]].begin(), comp[cand[ind^msk]].end());
		swap(l,r);
	}
	//print(l);
	//print(r);
	lo = 0, hi = r.size()-1;
	if (hi == lo)
		hi--;
	while (lo <= hi) {
		int mi = (lo+hi) / 2;
		vi tmp;
		for (int i = 0; i <= mi; i++)
			tmp.pb(r[i]);
		int la[l.size()], ra[tmp.size()];
		copy(l.begin(), l.end(), la);
		copy(tmp.begin(), tmp.end(), ra);
		int fl = query((int)l.size(),(int)tmp.size(), la, ra);
		if (fl == 1)
			hi = mi - 1;
		else
			lo = mi + 1;
	}
	ret.se = r[lo];
	//cerr << ret.fi << " " << ret.se << "\n-----------------------\n";
	return ret;
}


void run(int N) {
	for (int i = 1; i <= N; i++)
		par[i] = i, sz[i] = 1, comp[i] = {i};
	for (int t = 1; t < N; t++) {
		int a, b;
		tie(a, b) = solve(N);
		setRoad(a, b);
		join(a, b);
	}
}

Compilation message

icc.cpp: In function 'void print(std::vector<int>)':
icc.cpp:51:2: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
  for (int i: cur) cerr << i << " "; cerr << "\n";
  ^~~
icc.cpp:51:37: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
  for (int i: cur) cerr << i << " "; cerr << "\n";
                                     ^~~~

Subtask #1
7.0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Correct 9 ms 504 KB Ok! 114 queries used.
2 Correct 9 ms 504 KB Ok! 108 queries used.

Subtask #2
11.0 / 11.0

# 결과 실행 시간 메모리 Grader output
1 Correct 48 ms 504 KB Ok! 594 queries used.
2 Correct 43 ms 596 KB Ok! 524 queries used.
3 Correct 44 ms 564 KB Ok! 554 queries used.

Subtask #3
22.0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Correct 152 ms 564 KB Ok! 1443 queries used.
2 Correct 133 ms 568 KB Ok! 1221 queries used.
3 Correct 155 ms 600 KB Ok! 1528 queries used.
4 Correct 154 ms 576 KB Ok! 1450 queries used.

Subtask #4
21.0 / 21.0

# 결과 실행 시간 메모리 Grader output
1 Correct 150 ms 572 KB Ok! 1459 queries used.
2 Correct 152 ms 564 KB Ok! 1443 queries used.
3 Correct 159 ms 604 KB Ok! 1510 queries used.
4 Correct 153 ms 504 KB Ok! 1506 queries used.

Subtask #5
29.0 / 29.0

# 결과 실행 시간 메모리 Grader output
1 Correct 151 ms 504 KB Ok! 1452 queries used.
2 Correct 156 ms 580 KB Ok! 1484 queries used.
3 Correct 155 ms 604 KB Ok! 1448 queries used.
4 Correct 157 ms 632 KB Ok! 1507 queries used.
5 Correct 154 ms 568 KB Ok! 1497 queries used.
6 Correct 152 ms 660 KB Ok! 1474 queries used.

Subtask #6
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 154 ms 504 KB Ok! 1453 queries used.
2 Correct 149 ms 632 KB Ok! 1381 queries used.