Submission #733947

# Submission time Handle Problem Language Result Execution time Memory
733947 2023-05-01T12:19:37 Z vjudge1 City Mapping (NOI18_citymapping) C++17
57 / 100
118 ms 8908 KB
#include "citymapping.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr ll inf = 1e15;
void find_roads(int N, int Q, int A[], int B[], int W[]) {
	if (Q == 500000) {
		vector<tuple<ll, int, int>> T;
		for (int i = 0; i < N; i++) {
			for (int j = i + 1; j < N; j++) {
				T.emplace_back(get_distance(i + 1, j + 1), i, j);
			}
		}
		sort(T.begin(), T.end());
		vector<int> par(N);
		iota(par.begin(), par.end(), 0);
		function<int(int)> find = [&](int x) {
			return x == par[x] ? x : par[x] = find(par[x]);
		};
		int p = 0;
		for (int i = 0; i < T.size(); i++) {
			auto [z, x, y] = T[i];
			if (find(x) != find(y)) {
				A[p] = x + 1;
				B[p] = y + 1;
				W[p] = z;
				p++;
				par[find(x)] = find(y);
			}
		}
		return;
	}
	if (Q == 12000) {
		int x = -1;
		ll mx = 0;
		for (int i = 1; i < N; i++) {
			ll s = get_distance(1, i + 1);
			if (s > mx) {
				mx = s;
				x = i;
			}
		}
		vector<pair<ll, int>> p(N);
		for (int i = 0; i < N; i++) {
			p[i] = {get_distance(x + 1, i + 1), i};
		}
		sort(p.begin(), p.end());
		for (int i = 0; i < N - 1; i++) {
			A[i] = p[i].second + 1;
			B[i] = p[i + 1].second + 1;
			W[i] = p[i + 1].first - p[i].first;
		}
		return;
	}
    int r = -1;
    ll mx = 0;
    for (int i = 1; i < N; i++) {
		assert(Q--);
		ll s = get_distance(1, i + 1);
		if (s > mx) {
			mx = s;
			r = i;
		}
	}
	vector<ll> dis(N);
	for (int i = 0; i < N; i++) {
		assert(Q--);
		dis[i] = i == r ? 0 : get_distance(r + 1, i + 1);
	}
	vector<int> p(N);
	iota(p.begin(), p.end(), 0);
	sort(p.begin(), p.end(), [&](int i, int j) {
		return dis[i] < dis[j];
	});
	vector<vector<int>> d(N);
	d[0].push_back(r);
	for (int i = 1; i < N; i++) {
		int v = p[i];
		vector<bool> f(N);
		int lo = 0, hi = N - 1;
		while (lo < hi) {
			int mi = (lo + hi + 1) / 2;
			bool ok = 0;
			for (int x : d[mi]) {
				assert(Q--);
				if (dis[v] - dis[x] == get_distance(x + 1, v + 1)) {
					f[x] = 1;
					ok = 1;
				}
			}
			if (ok) {
				lo = mi;
			} else {
				hi = mi - 1;
			}
		}
		int u = -1;
		for (int x : d[lo]) {
			if (f[x]) {
				u = x;
			}
		}
		if (u == -1) {
			for (int x : d[lo]) {
				assert(Q--);
				if (dis[v] - dis[x] == get_distance(x + 1, v + 1)) {
					u = x;
					break;
				}
			}
		}
		A[i - 1] = u + 1;
		B[i - 1] = v + 1;
		W[i - 1] = dis[v] - dis[u];
		d[lo + 1].push_back(v);
	}
}

Compilation message

citymapping.cpp: In function 'void find_roads(int, int, int*, int*, int*)':
citymapping.cpp:21:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::tuple<long long int, int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   21 |   for (int i = 0; i < T.size(); i++) {
      |                   ~~^~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 110 ms 8756 KB Correct: 498501 out of 500000 queries used.
2 Correct 108 ms 8708 KB Correct: 499500 out of 500000 queries used.
3 Correct 102 ms 8776 KB Correct: 492528 out of 500000 queries used.
4 Correct 96 ms 8832 KB Correct: 494515 out of 500000 queries used.
5 Correct 118 ms 8732 KB Correct: 498501 out of 500000 queries used.
# Verdict Execution time Memory Grader output
1 Correct 110 ms 8756 KB Correct: 498501 out of 500000 queries used.
2 Correct 108 ms 8708 KB Correct: 499500 out of 500000 queries used.
3 Correct 102 ms 8776 KB Correct: 492528 out of 500000 queries used.
4 Correct 96 ms 8832 KB Correct: 494515 out of 500000 queries used.
5 Correct 118 ms 8732 KB Correct: 498501 out of 500000 queries used.
6 Correct 98 ms 8820 KB Correct: 495510 out of 500000 queries used.
7 Correct 106 ms 8908 KB Correct: 497503 out of 500000 queries used.
8 Correct 97 ms 8728 KB Correct: 497503 out of 500000 queries used.
9 Correct 93 ms 8768 KB Correct: 495510 out of 500000 queries used.
10 Correct 106 ms 8724 KB Correct: 496506 out of 500000 queries used.
# Verdict Execution time Memory Grader output
1 Correct 1 ms 468 KB Correct: 1981 out of 12000 queries used.
2 Correct 2 ms 468 KB Correct: 1985 out of 12000 queries used.
3 Correct 1 ms 468 KB Correct: 1999 out of 12000 queries used.
4 Correct 1 ms 468 KB Correct: 1985 out of 12000 queries used.
5 Correct 1 ms 468 KB Correct: 1981 out of 12000 queries used.
# Verdict Execution time Memory Grader output
1 Correct 1 ms 468 KB Correct: 1981 out of 12000 queries used.
2 Correct 2 ms 468 KB Correct: 1985 out of 12000 queries used.
3 Correct 1 ms 468 KB Correct: 1999 out of 12000 queries used.
4 Correct 1 ms 468 KB Correct: 1985 out of 12000 queries used.
5 Correct 1 ms 468 KB Correct: 1981 out of 12000 queries used.
6 Correct 1 ms 468 KB Correct: 1995 out of 12000 queries used.
7 Correct 1 ms 468 KB Correct: 1991 out of 12000 queries used.
8 Correct 1 ms 468 KB Correct: 1999 out of 12000 queries used.
9 Correct 1 ms 468 KB Correct: 1993 out of 12000 queries used.
10 Correct 1 ms 468 KB Correct: 1987 out of 12000 queries used.
# Verdict Execution time Memory Grader output
1 Correct 110 ms 8756 KB Correct: 498501 out of 500000 queries used.
2 Correct 108 ms 8708 KB Correct: 499500 out of 500000 queries used.
3 Correct 102 ms 8776 KB Correct: 492528 out of 500000 queries used.
4 Correct 96 ms 8832 KB Correct: 494515 out of 500000 queries used.
5 Correct 118 ms 8732 KB Correct: 498501 out of 500000 queries used.
6 Correct 98 ms 8820 KB Correct: 495510 out of 500000 queries used.
7 Correct 106 ms 8908 KB Correct: 497503 out of 500000 queries used.
8 Correct 97 ms 8728 KB Correct: 497503 out of 500000 queries used.
9 Correct 93 ms 8768 KB Correct: 495510 out of 500000 queries used.
10 Correct 106 ms 8724 KB Correct: 496506 out of 500000 queries used.
11 Correct 1 ms 468 KB Correct: 1981 out of 12000 queries used.
12 Correct 2 ms 468 KB Correct: 1985 out of 12000 queries used.
13 Correct 1 ms 468 KB Correct: 1999 out of 12000 queries used.
14 Correct 1 ms 468 KB Correct: 1985 out of 12000 queries used.
15 Correct 1 ms 468 KB Correct: 1981 out of 12000 queries used.
16 Correct 1 ms 468 KB Correct: 1995 out of 12000 queries used.
17 Correct 1 ms 468 KB Correct: 1991 out of 12000 queries used.
18 Correct 1 ms 468 KB Correct: 1999 out of 12000 queries used.
19 Correct 1 ms 468 KB Correct: 1993 out of 12000 queries used.
20 Correct 1 ms 468 KB Correct: 1987 out of 12000 queries used.
21 Correct 2 ms 468 KB Correct: 7009 out of 25000 queries used.
22 Correct 2 ms 468 KB Correct: 10929 out of 25000 queries used.
23 Correct 2 ms 468 KB Correct: 10695 out of 25000 queries used.
24 Correct 2 ms 468 KB Correct: 11061 out of 25000 queries used.
25 Runtime error 3 ms 724 KB Execution killed with signal 6
26 Halted 0 ms 0 KB -