답안 #870842

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
870842	2023-11-09T09:39:09 Z	marvinthang	즐거운 행로 (APIO20_fun)	C++17	100 / 100	183 ms	21704 KB

/*************************************
*    author: marvinthang             *
*    created: 09.11.2023 14:50:12    *
*************************************/

#include "fun.h"
#include <bits/stdc++.h>

using namespace std;

#define                  fi  first
#define                  se  second
#define                left  ___left
#define               right  ___right
#define                TIME  (1.0 * clock() / CLOCKS_PER_SEC)
#define             MASK(i)  (1LL << (i))
#define           BIT(x, i)  ((x) >> (i) & 1)
#define  __builtin_popcount  __builtin_popcountll
#define              ALL(v)  (v).begin(), (v).end()
#define           REP(i, n)  for (int i = 0, _n = (n); i < _n; ++i)
#define          REPD(i, n)  for (int i = (n); i-- > 0; )
#define        FOR(i, a, b)  for (int i = (a), _b = (b); i < _b; ++i) 
#define       FORD(i, b, a)  for (int i = (b), _a = (a); --i >= _a; ) 
#define       FORE(i, a, b)  for (int i = (a), _b = (b); i <= _b; ++i) 
#define      FORDE(i, b, a)  for (int i = (b), _a = (a); i >= _a; --i) 
#define        scan_op(...)  istream & operator >> (istream &in, __VA_ARGS__ &u)
#define       print_op(...)  ostream & operator << (ostream &out, const __VA_ARGS__ &u)
#ifdef LOCAL
		#include "debug.h"
#else
		#define file(name) if (fopen(name".inp", "r")) { freopen(name".inp", "r", stdin); freopen(name".out", "w", stdout); }
		#define DB(...) 23
		#define db(...) 23
		#define debug(...) 23
#endif

template <class U, class V> scan_op(pair <U, V>)  { return in >> u.first >> u.second; }
template <class T> scan_op(vector <T>)  { for (size_t i = 0; i < u.size(); ++i) in >> u[i]; return in; }
template <class U, class V> print_op(pair <U, V>)  { return out << '(' << u.first << ", " << u.second << ')'; }
template <size_t i, class T> ostream & print_tuple_utils(ostream &out, const T &tup) { if constexpr(i == tuple_size<T>::value) return out << ")";  else return print_tuple_utils<i + 1, T>(out << (i ? ", " : "(") << get<i>(tup), tup); }
template <class ...U> print_op(tuple<U...>) { return print_tuple_utils<0, tuple<U...>>(out, u); }
template <class Con, class = decltype(begin(declval<Con>()))> typename enable_if <!is_same<Con, string>::value, ostream&>::type operator << (ostream &out, const Con &con) { out << '{'; for (__typeof(con.begin()) it = con.begin(); it != con.end(); ++it) out << (it == con.begin() ? "" : ", ") << *it; return out << '}'; }

// end of template

vector <int> createFunTour(int n, int q) {
	int min_sz = n, cen = 0;
	FOR(u, 1, n) {
		int sz_u = attractionsBehind(0, u);
		if (sz_u > n / 2 && sz_u < min_sz) min_sz = sz_u, cen = u;
	}
	vector <int> dist(n), child;
	REP(i, n) if (i != cen) {
		dist[i] = hoursRequired(cen, i);
		if (dist[i] == 1) child.push_back(i);
	}
	vector <int> comps(n, child.size() - 1);
	REP(i, child.size() - 1) {
		comps[child[i]] = i;
		REP(u, n) if (u != cen && comps[u] == child.size() - 1 && hoursRequired(u, child[i]) + 1 == dist[u]) comps[u] = i;
	}
	vector <int> sz(3);
	REP(i, n) if (i != cen) ++sz[comps[i]];
	int mi = min_element(ALL(sz)) - sz.begin();
	comps[cen] = mi;
	++sz[mi];
	vector <vector <int>> List(3);
	REP(i, n) List[comps[i]].push_back(i);
	REP(i, 3) sort(ALL(List[i]), [&] (int u, int v) { return dist[u] < dist[v]; });
	int last = -1;
	vector <int> res;
	FORDE(sum, n, 1) {
		bool ok = false;
		REP(i, 3) if (sz[i] + sz[i] == sum) {
			vector <int> new_list;
			int f = 1;
			if (last >= 0 && last != i) {
				int k = 3 - i - last;
				if (dist[List[last].back()] >= dist[List[3 - i - last].back()]) f = 0;
			}
			REP(j, 3) if (j != i) new_list.insert(new_list.end(), ALL(List[j]));
			sort(ALL(new_list), [&] (int u, int v) { return dist[u] < dist[v]; });
			REPD(t, sz[i]) {
				if (f) {
					res.push_back(new_list[t]);
					res.push_back(List[i][t]);
				} else {
					res.push_back(List[i][t]);
					res.push_back(new_list[t]);
				}
			}
			ok = true;
			break;
		}
		if (ok) break;
		int ma = -1;
		REP(i, 3) if (i != last && !List[i].empty() && (ma < 0 || dist[List[ma].back()] < dist[List[i].back()])) ma = i;
		res.push_back(List[ma].back());
		List[ma].pop_back();
		last = ma;
		--sz[ma];
	}
	return res;
}

Compilation message

fun.cpp: In function 'std::vector<int> createFunTour(int, int)':
fun.cpp:60:38: warning: comparison of integer expressions of different signedness: '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   60 |   REP(u, n) if (u != cen && comps[u] == child.size() - 1 && hoursRequired(u, child[i]) + 1 == dist[u]) comps[u] = i;
fun.cpp:78:9: warning: unused variable 'k' [-Wunused-variable]
   78 |     int k = 3 - i - last;
      |         ^

Subtask #1
10.0 / 10.0

#	결과	실행 시간	메모리	Grader output
1	Correct	0 ms	344 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	0 ms	348 KB	Output is correct
9	Correct	0 ms	348 KB	Output is correct
10	Correct	0 ms	348 KB	Output is correct
11	Correct	0 ms	348 KB	Output is correct
12	Correct	0 ms	348 KB	Output is correct
13	Correct	0 ms	348 KB	Output is correct
14	Correct	0 ms	348 KB	Output is correct
15	Correct	0 ms	348 KB	Output is correct
16	Correct	1 ms	348 KB	Output is correct
17	Correct	0 ms	348 KB	Output is correct
18	Correct	0 ms	348 KB	Output is correct
19	Correct	0 ms	348 KB	Output is correct
20	Correct	0 ms	348 KB	Output is correct

Subtask #2
16.0 / 16.0

#	결과	실행 시간	메모리	Grader output
1	Correct	0 ms	344 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	0 ms	348 KB	Output is correct
9	Correct	0 ms	348 KB	Output is correct
10	Correct	0 ms	348 KB	Output is correct
11	Correct	0 ms	348 KB	Output is correct
12	Correct	0 ms	348 KB	Output is correct
13	Correct	0 ms	348 KB	Output is correct
14	Correct	0 ms	348 KB	Output is correct
15	Correct	0 ms	348 KB	Output is correct
16	Correct	1 ms	348 KB	Output is correct
17	Correct	0 ms	348 KB	Output is correct
18	Correct	0 ms	348 KB	Output is correct
19	Correct	0 ms	348 KB	Output is correct
20	Correct	0 ms	348 KB	Output is correct
21	Correct	0 ms	348 KB	Output is correct
22	Correct	0 ms	348 KB	Output is correct
23	Correct	0 ms	348 KB	Output is correct
24	Correct	0 ms	348 KB	Output is correct
25	Correct	0 ms	348 KB	Output is correct
26	Correct	0 ms	348 KB	Output is correct
27	Correct	0 ms	348 KB	Output is correct
28	Correct	0 ms	348 KB	Output is correct
29	Correct	1 ms	348 KB	Output is correct
30	Correct	1 ms	348 KB	Output is correct
31	Correct	0 ms	348 KB	Output is correct
32	Correct	1 ms	348 KB	Output is correct
33	Correct	0 ms	348 KB	Output is correct
34	Correct	1 ms	344 KB	Output is correct
35	Correct	1 ms	344 KB	Output is correct
36	Correct	1 ms	348 KB	Output is correct
37	Correct	0 ms	348 KB	Output is correct
38	Correct	1 ms	348 KB	Output is correct
39	Correct	0 ms	348 KB	Output is correct
40	Correct	1 ms	348 KB	Output is correct
41	Correct	1 ms	348 KB	Output is correct

Subtask #3
21.0 / 21.0

#	결과	실행 시간	메모리	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	0 ms	348 KB	Output is correct
9	Correct	0 ms	348 KB	Output is correct
10	Correct	0 ms	348 KB	Output is correct
11	Correct	0 ms	348 KB	Output is correct
12	Correct	0 ms	348 KB	Output is correct
13	Correct	0 ms	348 KB	Output is correct
14	Correct	0 ms	348 KB	Output is correct
15	Correct	0 ms	348 KB	Output is correct
16	Correct	1 ms	344 KB	Output is correct
17	Correct	1 ms	348 KB	Output is correct
18	Correct	86 ms	15552 KB	Output is correct
19	Correct	1 ms	344 KB	Output is correct
20	Correct	1 ms	616 KB	Output is correct
21	Correct	2 ms	604 KB	Output is correct
22	Correct	4 ms	1368 KB	Output is correct
23	Correct	8 ms	2140 KB	Output is correct
24	Correct	11 ms	2908 KB	Output is correct
25	Correct	48 ms	9076 KB	Output is correct

Subtask #4
19.0 / 19.0

#	결과	실행 시간	메모리	Grader output
1	Correct	0 ms	348 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	1 ms	348 KB	Output is correct
9	Correct	1 ms	348 KB	Output is correct
10	Correct	0 ms	348 KB	Output is correct
11	Correct	1 ms	348 KB	Output is correct
12	Correct	0 ms	348 KB	Output is correct
13	Correct	1 ms	348 KB	Output is correct
14	Correct	6 ms	1628 KB	Output is correct
15	Correct	93 ms	15308 KB	Output is correct
16	Correct	105 ms	15104 KB	Output is correct
17	Correct	17 ms	3928 KB	Output is correct
18	Correct	42 ms	7068 KB	Output is correct
19	Correct	70 ms	12068 KB	Output is correct
20	Correct	3 ms	860 KB	Output is correct

Subtask #5
34.0 / 34.0

#	결과	실행 시간	메모리	Grader output
1	Correct	0 ms	344 KB	Output is correct
2	Correct	0 ms	348 KB	Output is correct
3	Correct	0 ms	348 KB	Output is correct
4	Correct	0 ms	348 KB	Output is correct
5	Correct	0 ms	348 KB	Output is correct
6	Correct	0 ms	348 KB	Output is correct
7	Correct	0 ms	348 KB	Output is correct
8	Correct	0 ms	348 KB	Output is correct
9	Correct	0 ms	348 KB	Output is correct
10	Correct	0 ms	348 KB	Output is correct
11	Correct	0 ms	348 KB	Output is correct
12	Correct	0 ms	348 KB	Output is correct
13	Correct	0 ms	348 KB	Output is correct
14	Correct	0 ms	348 KB	Output is correct
15	Correct	0 ms	348 KB	Output is correct
16	Correct	1 ms	348 KB	Output is correct
17	Correct	0 ms	348 KB	Output is correct
18	Correct	0 ms	348 KB	Output is correct
19	Correct	0 ms	348 KB	Output is correct
20	Correct	0 ms	348 KB	Output is correct
21	Correct	0 ms	348 KB	Output is correct
22	Correct	0 ms	348 KB	Output is correct
23	Correct	0 ms	348 KB	Output is correct
24	Correct	0 ms	348 KB	Output is correct
25	Correct	0 ms	348 KB	Output is correct
26	Correct	0 ms	348 KB	Output is correct
27	Correct	0 ms	348 KB	Output is correct
28	Correct	0 ms	348 KB	Output is correct
29	Correct	1 ms	348 KB	Output is correct
30	Correct	1 ms	348 KB	Output is correct
31	Correct	0 ms	348 KB	Output is correct
32	Correct	1 ms	348 KB	Output is correct
33	Correct	0 ms	348 KB	Output is correct
34	Correct	1 ms	344 KB	Output is correct
35	Correct	1 ms	344 KB	Output is correct
36	Correct	1 ms	348 KB	Output is correct
37	Correct	0 ms	348 KB	Output is correct
38	Correct	1 ms	348 KB	Output is correct
39	Correct	0 ms	348 KB	Output is correct
40	Correct	1 ms	348 KB	Output is correct
41	Correct	1 ms	348 KB	Output is correct
42	Correct	1 ms	344 KB	Output is correct
43	Correct	1 ms	348 KB	Output is correct
44	Correct	86 ms	15552 KB	Output is correct
45	Correct	1 ms	344 KB	Output is correct
46	Correct	1 ms	616 KB	Output is correct
47	Correct	2 ms	604 KB	Output is correct
48	Correct	4 ms	1368 KB	Output is correct
49	Correct	8 ms	2140 KB	Output is correct
50	Correct	11 ms	2908 KB	Output is correct
51	Correct	48 ms	9076 KB	Output is correct
52	Correct	1 ms	348 KB	Output is correct
53	Correct	6 ms	1628 KB	Output is correct
54	Correct	93 ms	15308 KB	Output is correct
55	Correct	105 ms	15104 KB	Output is correct
56	Correct	17 ms	3928 KB	Output is correct
57	Correct	42 ms	7068 KB	Output is correct
58	Correct	70 ms	12068 KB	Output is correct
59	Correct	3 ms	860 KB	Output is correct
60	Correct	122 ms	16844 KB	Output is correct
61	Correct	137 ms	19148 KB	Output is correct
62	Correct	105 ms	16844 KB	Output is correct
63	Correct	133 ms	20744 KB	Output is correct
64	Correct	122 ms	20612 KB	Output is correct
65	Correct	139 ms	16580 KB	Output is correct
66	Correct	150 ms	18892 KB	Output is correct
67	Correct	128 ms	20728 KB	Output is correct
68	Correct	169 ms	20080 KB	Output is correct
69	Correct	165 ms	21188 KB	Output is correct
70	Correct	105 ms	16700 KB	Output is correct
71	Correct	121 ms	20356 KB	Output is correct
72	Correct	141 ms	19148 KB	Output is correct
73	Correct	134 ms	20168 KB	Output is correct
74	Correct	183 ms	21284 KB	Output is correct
75	Correct	133 ms	16984 KB	Output is correct
76	Correct	131 ms	20612 KB	Output is correct
77	Correct	142 ms	20868 KB	Output is correct
78	Correct	130 ms	19984 KB	Output is correct
79	Correct	166 ms	21332 KB	Output is correct
80	Correct	147 ms	21704 KB	Output is correct
81	Correct	127 ms	16592 KB	Output is correct
82	Correct	138 ms	16836 KB	Output is correct
83	Correct	127 ms	16664 KB	Output is correct
84	Correct	47 ms	7092 KB	Output is correct

답안 #870842

Compilation message

Subtask #1 10.0 / 10.0

Subtask #2 16.0 / 16.0

Subtask #3 21.0 / 21.0

Subtask #4 19.0 / 19.0

Subtask #5 34.0 / 34.0

Subtask #1
10.0 / 10.0

Subtask #2
16.0 / 16.0

Subtask #3
21.0 / 21.0

Subtask #4
19.0 / 19.0

Subtask #5
34.0 / 34.0