#include <bits/stdc++.h>
#include "longesttrip.h"
using namespace std;
int A[300][300];
int ask(vector<int> a, vector<int> b)
{
for (int x : a)
{
for (int y : b)
{
if (A[x][y]) return 1;
}
}
return 0;
}
vector<int> longest_trip(int n, int D)
{
for (int i = 0; i < n; i++)
{
A[i][i] = 1;
for (int j = i + 1; j < n; j++)
{
A[i][j] = A[j][i] = are_connected({i}, {j});
}
}
vector<int> used(n, 0);
vector<int> v1, v2;
for (int i = 0; i < 3; i++)
{
for (int j = i + 1; j < 3; j++)
{
if (ask({i}, {j}))
{
v1.push_back(i), v1.push_back(j);
used[i] = used[j] = 1;
break;
}
}
if (!v1.empty()) break;
}
for (int i = 0; i < n; i++)
{
if (used[i]) continue;
if (ask(v1, {i}) && ask(v2, {i}))
{
for (int x : v1) for (int y : v1) assert(ask({x}, {y}));
for (int x : v2) for (int y : v2) assert(ask({x}, {y}));
int id1 = -1, id2 = -1;
for (int j = 0; j < v1.size(); j++)
{
if (ask({v1[j]}, {i}))
{
id1 = j;
break;
}
}
for (int j = 0; j < v2.size(); j++)
{
if (ask({v2[j]}, {i}))
{
id2 = j;
break;
}
}
vector<int> nw;
for (int j = (id2 + 1) % v2.size(); 1; j = (j + 1) % v2.size())
{
nw.push_back(v2[j]);
if (j == id2) break;
}
nw.push_back(i), nw.push_back(v1[id1]);
for (int j = (id1 + 1) % v1.size(); j != id1; j = (j + 1) % v1.size())
{
nw.push_back(v1[j]);
}
v2.clear();
v1 = nw;
}
else if (ask(v1, {i}))
{
vector<int> nw;
int idx = -1;
for (int j = 0; j < v1.size(); j++)
{
if (ask({i}, {v1[j]}))
{
idx = j;
break;
}
}
nw.push_back(i), nw.push_back(v1[idx]);
for (int j = (idx + 1) % v1.size(); j != idx; j = (j + 1) % v1.size())
{
nw.push_back(v1[j]);
}
v1 = nw;
}
else if (ask(v2, {i}))
{
for (int x : v1)
{
for (int y : v1) assert(ask({x}, {y}));
}
vector<int> nw;
int idx = -1;
for (int j = 0; j < v2.size(); j++)
{
if (ask({i}, {v2[j]}))
{
idx = j;
break;
}
}
nw.push_back(i), nw.push_back(v2[idx]);
for (int j = (idx + 1) % v2.size(); j != idx; j = (j + 1) % v2.size())
{
nw.push_back(v2[j]);
}
v2 = nw;
}
else v2.push_back(i);
}
vector<int> res;
if (max(v1.size(), v2.size()) == v1.size()) res = v1;
else if (max(v1.size(), v2.size()) == v2.size()) res = v2;
int prev = res[0], f = 1;
for (int i = 1; i < res.size(); i++)
{
f &= A[prev][res[i]];
prev = res[i];
}
return res;
};
Compilation message
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:53:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
53 | for (int j = 0; j < v1.size(); j++)
| ~~^~~~~~~~~~~
longesttrip.cpp:61:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
61 | for (int j = 0; j < v2.size(); j++)
| ~~^~~~~~~~~~~
longesttrip.cpp:87:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
87 | for (int j = 0; j < v1.size(); j++)
| ~~^~~~~~~~~~~
longesttrip.cpp:110:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
110 | for (int j = 0; j < v2.size(); j++)
| ~~^~~~~~~~~~~
longesttrip.cpp:131:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
131 | for (int i = 1; i < res.size(); i++)
| ~~^~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
149 ms |
724 KB |
Incorrect |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
344 KB |
Output is correct |
2 |
Correct |
21 ms |
344 KB |
Output is correct |
3 |
Correct |
110 ms |
600 KB |
Output is correct |
4 |
Correct |
312 ms |
564 KB |
Output is correct |
5 |
Correct |
676 ms |
716 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
344 KB |
Output is correct |
2 |
Correct |
20 ms |
344 KB |
Output is correct |
3 |
Correct |
110 ms |
600 KB |
Output is correct |
4 |
Correct |
317 ms |
580 KB |
Output is correct |
5 |
Correct |
689 ms |
720 KB |
Output is correct |
6 |
Correct |
6 ms |
344 KB |
Output is correct |
7 |
Correct |
23 ms |
600 KB |
Output is correct |
8 |
Correct |
111 ms |
600 KB |
Output is correct |
9 |
Correct |
257 ms |
532 KB |
Output is correct |
10 |
Correct |
686 ms |
720 KB |
Output is correct |
11 |
Correct |
665 ms |
724 KB |
Output is correct |
12 |
Correct |
633 ms |
720 KB |
Output is correct |
13 |
Correct |
689 ms |
720 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
344 KB |
Output is correct |
2 |
Correct |
21 ms |
344 KB |
Output is correct |
3 |
Correct |
112 ms |
600 KB |
Output is correct |
4 |
Correct |
336 ms |
596 KB |
Output is correct |
5 |
Correct |
682 ms |
848 KB |
Output is correct |
6 |
Correct |
6 ms |
344 KB |
Output is correct |
7 |
Correct |
20 ms |
344 KB |
Output is correct |
8 |
Correct |
108 ms |
600 KB |
Output is correct |
9 |
Correct |
248 ms |
592 KB |
Output is correct |
10 |
Correct |
678 ms |
716 KB |
Output is correct |
11 |
Correct |
699 ms |
856 KB |
Output is correct |
12 |
Correct |
711 ms |
848 KB |
Output is correct |
13 |
Correct |
684 ms |
720 KB |
Output is correct |
14 |
Correct |
7 ms |
344 KB |
Output is correct |
15 |
Incorrect |
4 ms |
344 KB |
Incorrect |
16 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
344 KB |
Output is correct |
2 |
Correct |
21 ms |
344 KB |
Output is correct |
3 |
Partially correct |
117 ms |
600 KB |
Output is partially correct |
4 |
Partially correct |
334 ms |
568 KB |
Output is partially correct |
5 |
Partially correct |
666 ms |
852 KB |
Output is partially correct |
6 |
Correct |
7 ms |
344 KB |
Output is correct |
7 |
Correct |
23 ms |
344 KB |
Output is correct |
8 |
Partially correct |
121 ms |
496 KB |
Output is partially correct |
9 |
Partially correct |
249 ms |
536 KB |
Output is partially correct |
10 |
Partially correct |
663 ms |
720 KB |
Output is partially correct |
11 |
Partially correct |
643 ms |
856 KB |
Output is partially correct |
12 |
Partially correct |
720 ms |
720 KB |
Output is partially correct |
13 |
Partially correct |
708 ms |
724 KB |
Output is partially correct |
14 |
Correct |
6 ms |
344 KB |
Output is correct |
15 |
Incorrect |
4 ms |
600 KB |
Incorrect |
16 |
Halted |
0 ms |
0 KB |
- |