#include "citymapping.h"
#include <bits/stdc++.h>
namespace operators
{
template <typename T1, typename T2>
std::istream &operator>>(std::istream &in, std::pair<T1, T2> &x)
{
in >> x.first >> x.second;
return in;
}
template <typename T1, typename T2>
std::ostream &operator<<(std::ostream &out, std::pair<T1, T2> x)
{
out << x.first << " " << x.second;
return out;
}
template <typename T1>
std::istream &operator>>(std::istream &in, std::vector<T1> &x)
{
for (auto &i : x)
in >> i;
return in;
}
template <typename T1>
std::ostream &operator<<(std::ostream &out, std::vector<T1> &x)
{
for (auto &i : x)
out << i << " ";
return out;
}
template <typename T1>
std::ostream &operator<<(std::ostream &out, std::set<T1> &x)
{
for (auto &i : x)
out << i << " ";
return out;
}
template <typename T1>
std::ostream &operator<<(std::ostream &out, std::multiset<T1> &x)
{
for (auto &i : x)
out << i << " ";
return out;
}
}
// name spaces
using namespace std;
using namespace operators;
// end of name spaces
// defines
#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
#define pii pair<ll, ll>
#define pll pair<ll, ll>
#define f first
#define s second
#define uint unsigned int
#define all(vc) vc.begin(), vc.end()
// end of defines
vector<vector<int>> tree;
vector<pair<pll, ll>> edges;
vector<ll> D, sz;
void dfs(int v)
{
sz[v] = 1;
for (auto tov : tree[v])
{
dfs(tov);
sz[v] += sz[tov];
}
}
void find_roads(int N, int Q, int A[], int B[], int W[])
{
tree.resize(N);
D.resize(N);
sz.resize(N);
vector<pii> target;
for (ll i = 1; i < N; ++i)
{
D[i] = (ll)get_distance(1, i + 1);
target.pb({D[i], i});
sort(all(target));
}
ll i = 0;
vector<ll> pr(N);
for (auto [d, v] : target)
{
vector<ll> used(N, 0);
dfs(0);
ll root = 0;
used[root] = 1;
while (true)
{
ll bf = root;
while (true)
{
ll to = -1;
for (auto tov : tree[root])
if (!used[tov] and (to == -1 or sz[to] < sz[tov]))
to = tov;
if (to == -1)
break;
root = to;
used[root] = 1;
}
if (root == bf)
{
tree[root].pb(v);
A[i] = v + 1;
B[i] = root + 1;
W[i] = D[v] - D[root];
pr[v] = root;
++i;
break;
}
ll D1 = get_distance(root + 1, v + 1);
ll upTo = D1 - ((D1 + D[v] - D[root]) / 2);
if (D[v] == D[root] + D1)
continue;
while (upTo > 0 and root)
{
upTo -= D[root] - D[pr[root]];
if (upTo >= 0)
root = pr[root];
}
}
}
return;
}
// ////////////////////////////////////////////////////////
// // #include "citymapping.h"
// #include <bits/stdc++.h>
// using namespace std;
// static int N, Q, S, twok[1005][11], depth[1005], curQ, target = 6500;
// static long long dist[1005];
// static vector<pair<int, int>> adjlist[1005];
// static vector<pair<pair<int, int>, int>> edgelist, user_edgelist;
// static void dfs(int x, int p, long long d, int dep)
// {
// dist[x] = d;
// depth[x] = dep;
// twok[x][0] = p;
// for (int i = 1; i <= 10; i++)
// {
// if (twok[x][i - 1] == -1)
// break;
// twok[x][i] = twok[twok[x][i - 1]][i - 1];
// }
// for (int i = 0; i < (int)adjlist[x].size(); i++)
// {
// if (adjlist[x][i].first == p)
// continue;
// dfs(adjlist[x][i].first, x, d + adjlist[x][i].second, dep + 1);
// }
// }
// static int lca(int x, int y)
// {
// if (depth[x] > depth[y])
// swap(x, y);
// int dd = depth[y] - depth[x];
// for (int i = 0; i <= 10; i++)
// if (dd & (1 << i))
// y = twok[y][i];
// if (x == y)
// return x;
// for (int i = 10; i >= 0; i--)
// {
// if (twok[x][i] != twok[y][i])
// {
// x = twok[x][i];
// y = twok[y][i];
// }
// }
// return twok[x][0];
// }
// long long get_distance(int X, int Y)
// {
// if (X <= 0 || X > N || Y <= 0 || Y > N)
// {
// printf("Wrong Answer: get_distance() arguments out of range.\n");
// exit(0);
// }
// curQ++;
// if (curQ > Q)
// {
// printf("Wrong Answer: Too many calls to get_distance().\n");
// exit(0);
// }
// return dist[X - 1] + dist[Y - 1] - dist[lca(X - 1, Y - 1)] * 2;
// }
// int main()
// {
// if (scanf("%d%d%d", &N, &Q, &S) != 3)
// {
// printf("Wrong Input Format\n");
// exit(0);
// }
// for (int i = 0; i < N - 1; i++)
// {
// int a, b, w;
// if (scanf("%d%d%d", &a, &b, &w) != 3)
// {
// printf("Wrong Input Format\n");
// return 0;
// }
// a--;
// b--;
// adjlist[a].push_back(make_pair(b, w));
// adjlist[b].push_back(make_pair(a, w));
// edgelist.push_back(make_pair(make_pair(min(a, b) + 1, max(a, b) + 1), w));
// }
// sort(edgelist.begin(), edgelist.end());
// memset(twok, -1, sizeof(twok));
// dfs(0, -1, 0, 0);
// int A[N - 1], B[N - 1], W[N - 1];
// memset(A, 0, sizeof(A));
// memset(B, 0, sizeof(B));
// memset(W, 0, sizeof(W));
// find_roads(N, Q, A, B, W);
// for (int i = 0; i < N - 1; i++)
// {
// user_edgelist.push_back(make_pair(make_pair(min(A[i], B[i]), max(A[i], B[i])), W[i]));
// }
// sort(user_edgelist.begin(), user_edgelist.end());
// if (edgelist != user_edgelist)
// {
// printf("Wrong Answer: Reported list of edges differ from actual.\n");
// exit(0);
// }
// cout << "Correct";
// exit(0);
// if (S <= 4)
// {
// printf("Score: 100.00%%\nCorrect: %d out of %d queries used.\n", curQ, Q);
// exit(0);
// }
// else
// {
// if (curQ <= target)
// printf("Score: 100.00%%\nCorrect: %d out of %d queries used.\n", curQ, Q);
// else if (curQ >= 12000)
// printf("Score: %.2lf%%\nCorrect: %d out of %d queries used.\n", (10.0 - 10.0 * ((double)(curQ - 12000) / 13000)) / 43 * 100, curQ, Q);
// else
// printf("Score: %.2lf%%\nCorrect: %d out of %d queries used.\n", (40.0 - 30.0 * ((double)(curQ - target) / (12000 - target))) / 43 * 100, curQ, Q);
// }
// }
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
22 ms |
688 KB |
Correct: 2691 out of 500000 queries used. |
2 |
Correct |
20 ms |
604 KB |
Correct: 2421 out of 500000 queries used. |
3 |
Correct |
21 ms |
600 KB |
Correct: 4517 out of 500000 queries used. |
4 |
Correct |
21 ms |
604 KB |
Correct: 5567 out of 500000 queries used. |
5 |
Correct |
26 ms |
628 KB |
Correct: 3387 out of 500000 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
22 ms |
688 KB |
Correct: 2691 out of 500000 queries used. |
2 |
Correct |
20 ms |
604 KB |
Correct: 2421 out of 500000 queries used. |
3 |
Correct |
21 ms |
600 KB |
Correct: 4517 out of 500000 queries used. |
4 |
Correct |
21 ms |
604 KB |
Correct: 5567 out of 500000 queries used. |
5 |
Correct |
26 ms |
628 KB |
Correct: 3387 out of 500000 queries used. |
6 |
Correct |
18 ms |
604 KB |
Correct: 2009 out of 500000 queries used. |
7 |
Correct |
18 ms |
684 KB |
Correct: 2063 out of 500000 queries used. |
8 |
Correct |
15 ms |
604 KB |
Correct: 4244 out of 500000 queries used. |
9 |
Correct |
15 ms |
604 KB |
Correct: 5089 out of 500000 queries used. |
10 |
Correct |
17 ms |
604 KB |
Correct: 3054 out of 500000 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
19 ms |
604 KB |
Correct: 2086 out of 12000 queries used. |
2 |
Correct |
21 ms |
696 KB |
Correct: 2304 out of 12000 queries used. |
3 |
Correct |
24 ms |
604 KB |
Correct: 2457 out of 12000 queries used. |
4 |
Correct |
21 ms |
672 KB |
Correct: 2470 out of 12000 queries used. |
5 |
Correct |
19 ms |
600 KB |
Correct: 2240 out of 12000 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
19 ms |
604 KB |
Correct: 2086 out of 12000 queries used. |
2 |
Correct |
21 ms |
696 KB |
Correct: 2304 out of 12000 queries used. |
3 |
Correct |
24 ms |
604 KB |
Correct: 2457 out of 12000 queries used. |
4 |
Correct |
21 ms |
672 KB |
Correct: 2470 out of 12000 queries used. |
5 |
Correct |
19 ms |
600 KB |
Correct: 2240 out of 12000 queries used. |
6 |
Correct |
18 ms |
604 KB |
Correct: 2473 out of 12000 queries used. |
7 |
Correct |
18 ms |
604 KB |
Correct: 2382 out of 12000 queries used. |
8 |
Correct |
19 ms |
604 KB |
Correct: 2207 out of 12000 queries used. |
9 |
Correct |
18 ms |
680 KB |
Correct: 2235 out of 12000 queries used. |
10 |
Correct |
18 ms |
692 KB |
Correct: 2302 out of 12000 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
22 ms |
688 KB |
Correct: 2691 out of 500000 queries used. |
2 |
Correct |
20 ms |
604 KB |
Correct: 2421 out of 500000 queries used. |
3 |
Correct |
21 ms |
600 KB |
Correct: 4517 out of 500000 queries used. |
4 |
Correct |
21 ms |
604 KB |
Correct: 5567 out of 500000 queries used. |
5 |
Correct |
26 ms |
628 KB |
Correct: 3387 out of 500000 queries used. |
6 |
Correct |
18 ms |
604 KB |
Correct: 2009 out of 500000 queries used. |
7 |
Correct |
18 ms |
684 KB |
Correct: 2063 out of 500000 queries used. |
8 |
Correct |
15 ms |
604 KB |
Correct: 4244 out of 500000 queries used. |
9 |
Correct |
15 ms |
604 KB |
Correct: 5089 out of 500000 queries used. |
10 |
Correct |
17 ms |
604 KB |
Correct: 3054 out of 500000 queries used. |
11 |
Correct |
19 ms |
604 KB |
Correct: 2086 out of 12000 queries used. |
12 |
Correct |
21 ms |
696 KB |
Correct: 2304 out of 12000 queries used. |
13 |
Correct |
24 ms |
604 KB |
Correct: 2457 out of 12000 queries used. |
14 |
Correct |
21 ms |
672 KB |
Correct: 2470 out of 12000 queries used. |
15 |
Correct |
19 ms |
600 KB |
Correct: 2240 out of 12000 queries used. |
16 |
Correct |
18 ms |
604 KB |
Correct: 2473 out of 12000 queries used. |
17 |
Correct |
18 ms |
604 KB |
Correct: 2382 out of 12000 queries used. |
18 |
Correct |
19 ms |
604 KB |
Correct: 2207 out of 12000 queries used. |
19 |
Correct |
18 ms |
680 KB |
Correct: 2235 out of 12000 queries used. |
20 |
Correct |
18 ms |
692 KB |
Correct: 2302 out of 12000 queries used. |
21 |
Correct |
18 ms |
660 KB |
Correct: 2502 out of 25000 queries used. |
22 |
Correct |
18 ms |
680 KB |
Correct: 2071 out of 25000 queries used. |
23 |
Correct |
18 ms |
604 KB |
Correct: 2284 out of 25000 queries used. |
24 |
Correct |
19 ms |
600 KB |
Correct: 2036 out of 25000 queries used. |
25 |
Correct |
15 ms |
604 KB |
Correct: 4436 out of 25000 queries used. |
26 |
Correct |
15 ms |
604 KB |
Correct: 4358 out of 25000 queries used. |
27 |
Correct |
15 ms |
604 KB |
Correct: 4307 out of 25000 queries used. |
28 |
Correct |
15 ms |
604 KB |
Correct: 4417 out of 25000 queries used. |
29 |
Correct |
15 ms |
604 KB |
Correct: 4502 out of 25000 queries used. |
30 |
Correct |
16 ms |
856 KB |
Correct: 4442 out of 25000 queries used. |
31 |
Correct |
15 ms |
644 KB |
Correct: 5151 out of 25000 queries used. |
32 |
Correct |
22 ms |
860 KB |
Correct: 2223 out of 25000 queries used. |
33 |
Correct |
15 ms |
636 KB |
Correct: 5083 out of 25000 queries used. |
34 |
Correct |
14 ms |
600 KB |
Correct: 5158 out of 25000 queries used. |
35 |
Correct |
15 ms |
604 KB |
Correct: 5100 out of 25000 queries used. |
36 |
Correct |
15 ms |
604 KB |
Correct: 5118 out of 25000 queries used. |
37 |
Correct |
15 ms |
644 KB |
Correct: 5144 out of 25000 queries used. |
38 |
Correct |
14 ms |
604 KB |
Correct: 5102 out of 25000 queries used. |
39 |
Correct |
15 ms |
604 KB |
Correct: 5135 out of 25000 queries used. |
40 |
Correct |
15 ms |
604 KB |
Correct: 5168 out of 25000 queries used. |
41 |
Correct |
14 ms |
604 KB |
Correct: 5124 out of 25000 queries used. |
42 |
Correct |
15 ms |
600 KB |
Correct: 5203 out of 25000 queries used. |
43 |
Correct |
18 ms |
600 KB |
Correct: 2087 out of 25000 queries used. |
44 |
Correct |
14 ms |
604 KB |
Correct: 5116 out of 25000 queries used. |
45 |
Correct |
15 ms |
628 KB |
Correct: 5090 out of 25000 queries used. |
46 |
Correct |
15 ms |
656 KB |
Correct: 5068 out of 25000 queries used. |
47 |
Correct |
18 ms |
644 KB |
Correct: 5179 out of 25000 queries used. |
48 |
Correct |
14 ms |
604 KB |
Correct: 5135 out of 25000 queries used. |
49 |
Correct |
15 ms |
652 KB |
Correct: 5091 out of 25000 queries used. |
50 |
Correct |
15 ms |
632 KB |
Correct: 5105 out of 25000 queries used. |
51 |
Correct |
19 ms |
648 KB |
Correct: 5099 out of 25000 queries used. |
52 |
Correct |
15 ms |
624 KB |
Correct: 5128 out of 25000 queries used. |
53 |
Correct |
17 ms |
656 KB |
Correct: 5144 out of 25000 queries used. |
54 |
Correct |
18 ms |
604 KB |
Correct: 2333 out of 25000 queries used. |
55 |
Correct |
16 ms |
656 KB |
Correct: 5196 out of 25000 queries used. |
56 |
Correct |
20 ms |
636 KB |
Correct: 5141 out of 25000 queries used. |
57 |
Correct |
15 ms |
648 KB |
Correct: 5125 out of 25000 queries used. |
58 |
Correct |
15 ms |
604 KB |
Correct: 5115 out of 25000 queries used. |
59 |
Correct |
14 ms |
604 KB |
Correct: 5104 out of 25000 queries used. |
60 |
Correct |
18 ms |
616 KB |
Correct: 3041 out of 25000 queries used. |
61 |
Correct |
16 ms |
604 KB |
Correct: 3317 out of 25000 queries used. |
62 |
Correct |
20 ms |
1132 KB |
Correct: 2917 out of 25000 queries used. |
63 |
Correct |
17 ms |
604 KB |
Correct: 3317 out of 25000 queries used. |
64 |
Correct |
16 ms |
604 KB |
Correct: 3103 out of 25000 queries used. |
65 |
Correct |
19 ms |
604 KB |
Correct: 2067 out of 25000 queries used. |
66 |
Correct |
17 ms |
604 KB |
Correct: 3228 out of 25000 queries used. |
67 |
Correct |
17 ms |
856 KB |
Correct: 2018 out of 25000 queries used. |
68 |
Correct |
18 ms |
656 KB |
Correct: 2394 out of 25000 queries used. |
69 |
Correct |
18 ms |
660 KB |
Correct: 2451 out of 25000 queries used. |
70 |
Correct |
19 ms |
656 KB |
Correct: 2414 out of 25000 queries used. |