#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;
int dc_left(vector<int> a, vector<int> b) {
if (a.size() == 1)
return a.back();
vector<int> left(a.begin(), a.begin() + a.size() / 2);
vector<int> right(a.begin() + a.size() / 2, a.end());
if (are_connected(left, b))
return dc_left(left, b);
else
return dc_left(right, b);
}
int dc_right(vector<int> a, vector<int> b) {
if (b.size() == 1)
return b.back();
vector<int> left(b.begin(), b.begin() + b.size() / 2);
vector<int> right(b.begin() + b.size() / 2, b.end());
if (are_connected(a, left))
return dc_right(a, left);
else
return dc_right(a, right);
}
vector<int> longest_trip(int N, int D) {
vector<vector<int>> g(N);
vector<deque<int>> comp(N);
for (int i = 0; i < N; i++)
comp[i].push_back(i);
while (comp.size() > 2) {
int i = 0;
int j = 1;
int k = 2;
int u = comp[i].back();
int v = comp[j].back();
int w = comp[k].back();
if (are_connected({v}, {w})) {
swap(u, w);
swap(i, k);
} else if (are_connected({u}, {w})) {
swap(v, w);
swap(j, k);
}
g[u].push_back(v);
g[v].push_back(u);
reverse(comp[j].begin(), comp[j].end());
comp[i].insert(comp[i].end(), comp[j].begin(), comp[j].end());
comp.erase(comp.begin() + j);
}
if (comp[0].size() < comp[1].size()) {
swap(comp[0], comp[1]);
}
if (are_connected({comp[0].back()}, {comp[1].back()})) {
reverse(comp[1].begin(), comp[1].end());
comp[0].insert(comp[0].end(), comp[1].begin(), comp[1].end());
} else if (are_connected({comp[0].back()}, {comp[1].front()})) {
comp[0].insert(comp[0].end(), comp[1].begin(), comp[1].end());
} else if (are_connected({comp[0].front()}, {comp[1].front()})) {
reverse(comp[0].begin(), comp[0].end());
comp[0].insert(comp[0].end(), comp[1].begin(), comp[1].end());
} else if (are_connected({comp[0].front()}, {comp[1].back()})) {
reverse(comp[0].begin(), comp[0].end());
reverse(comp[1].begin(), comp[1].end());
comp[0].insert(comp[0].end(), comp[1].begin(), comp[1].end());
} else if (are_connected(vector<int>(comp[0].begin(), comp[0].end()),
vector<int>(comp[1].begin(), comp[1].end()))) {
// two cycles, connected
int l = dc_left(vector<int>(comp[0].begin(), comp[0].end()),
vector<int>(comp[1].begin(), comp[1].end()));
int r = dc_right({l}, vector<int>(comp[1].begin(), comp[1].end()));
int l_idx = find(comp[0].begin(), comp[0].end(), l) - comp[0].begin();
int r_idx = find(comp[1].begin(), comp[1].end(), r) - comp[1].begin();
vector<int> ret;
for (int i = 0; i < comp[0].size(); i++)
ret.push_back(comp[0][(l_idx + i) % comp[0].size()]);
for (int i = 0; i < comp[1].size(); i++)
ret.push_back(comp[1][(r_idx + i) % comp[1].size()]);
return ret;
}
return vector<int>(comp[0].begin(), comp[0].end());
}
Compilation message
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:76:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
76 | for (int i = 0; i < comp[0].size(); i++)
| ~~^~~~~~~~~~~~~~~~
longesttrip.cpp:78:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
78 | for (int i = 0; i < comp[1].size(); i++)
| ~~^~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
2 ms |
600 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
344 KB |
Output is correct |
2 |
Correct |
5 ms |
344 KB |
Output is correct |
3 |
Correct |
8 ms |
344 KB |
Output is correct |
4 |
Correct |
6 ms |
344 KB |
Output is correct |
5 |
Correct |
7 ms |
600 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
344 KB |
Output is correct |
2 |
Correct |
6 ms |
344 KB |
Output is correct |
3 |
Correct |
7 ms |
344 KB |
Output is correct |
4 |
Correct |
8 ms |
520 KB |
Output is correct |
5 |
Correct |
6 ms |
600 KB |
Output is correct |
6 |
Correct |
7 ms |
344 KB |
Output is correct |
7 |
Correct |
6 ms |
344 KB |
Output is correct |
8 |
Correct |
6 ms |
344 KB |
Output is correct |
9 |
Correct |
6 ms |
344 KB |
Output is correct |
10 |
Correct |
7 ms |
600 KB |
Output is correct |
11 |
Correct |
7 ms |
600 KB |
Output is correct |
12 |
Correct |
6 ms |
600 KB |
Output is correct |
13 |
Correct |
7 ms |
600 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
344 KB |
Output is correct |
2 |
Correct |
4 ms |
344 KB |
Output is correct |
3 |
Correct |
6 ms |
344 KB |
Output is correct |
4 |
Correct |
6 ms |
520 KB |
Output is correct |
5 |
Correct |
6 ms |
600 KB |
Output is correct |
6 |
Correct |
7 ms |
344 KB |
Output is correct |
7 |
Correct |
5 ms |
340 KB |
Output is correct |
8 |
Correct |
8 ms |
344 KB |
Output is correct |
9 |
Correct |
6 ms |
344 KB |
Output is correct |
10 |
Correct |
7 ms |
600 KB |
Output is correct |
11 |
Correct |
8 ms |
600 KB |
Output is correct |
12 |
Correct |
7 ms |
600 KB |
Output is correct |
13 |
Correct |
7 ms |
600 KB |
Output is correct |
14 |
Correct |
16 ms |
344 KB |
Output is correct |
15 |
Correct |
10 ms |
344 KB |
Output is correct |
16 |
Incorrect |
2 ms |
340 KB |
Incorrect |
17 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
344 KB |
Output is correct |
2 |
Correct |
6 ms |
344 KB |
Output is correct |
3 |
Correct |
6 ms |
344 KB |
Output is correct |
4 |
Correct |
9 ms |
344 KB |
Output is correct |
5 |
Correct |
7 ms |
600 KB |
Output is correct |
6 |
Correct |
11 ms |
344 KB |
Output is correct |
7 |
Correct |
8 ms |
344 KB |
Output is correct |
8 |
Correct |
8 ms |
344 KB |
Output is correct |
9 |
Correct |
10 ms |
344 KB |
Output is correct |
10 |
Correct |
7 ms |
600 KB |
Output is correct |
11 |
Correct |
7 ms |
600 KB |
Output is correct |
12 |
Correct |
7 ms |
600 KB |
Output is correct |
13 |
Correct |
7 ms |
600 KB |
Output is correct |
14 |
Correct |
12 ms |
344 KB |
Output is correct |
15 |
Correct |
11 ms |
344 KB |
Output is correct |
16 |
Incorrect |
1 ms |
344 KB |
Incorrect |
17 |
Halted |
0 ms |
0 KB |
- |