Submission #958788

# Submission time Handle Problem Language Result Execution time Memory
958788 2024-04-06T16:21:16 Z nguyentunglam Longest Trip (IOI23_longesttrip) C++17
85 / 100
13 ms 952 KB
#include "longesttrip.h"
#ifdef ngu
#include "grader.cpp"
#endif // ngu
#include<bits/stdc++.h>
using namespace std;

pair<int, int> Find (vector<int> a, vector<int> b) {
  if (!are_connected(a, b)) return make_pair(-1, -1);
  int l = 0, r = a.size() - 1, lastx = -1, lasty = -1;
  while (l <= r) {
    int mid = l + r >> 1;
    vector<int> ask;
    for(int i = 0; i <= mid; i++) ask.push_back(a[i]);
    if (are_connected(ask, b)) {
      lastx = mid;
      r = mid - 1;
    } else l = mid + 1;
  }
  l = 0, r = b.size() - 1;
  while (l <= r) {
    int mid = l + r >> 1;
    vector<int> ask;
    for(int i = 0; i <= mid; i++) ask.push_back(b[i]);
    if (are_connected(ask, {a[lastx]})) {
      lasty = mid;
      r = mid - 1;
    } else l = mid + 1;
  }
  assert(are_connected({a[lastx]}, {b[lasty]}));
//  if(!are_connected({a[lastx]}, {b[lasty]})) {
//    cout << endl;
//    for(int &j : a) cout << j << " "; cout << endl;
//    for(int &j : b) cout << j << " "; cout << endl;
//    cout << lastx << " " << lasty << endl;
//    exit(0);
//  }
  return make_pair(lastx, lasty);
}

vector<int> rev (vector<int> a) {
  reverse(a.begin(), a.end());
  return a;
}

vector<int> sum (vector<int> a, vector<int> b) {
  for(int &j : b) a.push_back(j);
  return a;
}

std::vector<int> longest_trip(int n, int d)
{
    if (d == 1) {
      vector<int> a, b;
      a.push_back(0);
      b.push_back(1);
      for(int i = 2; i < n; i++) {
        if (are_connected({a.back()}, {i})) a.push_back(i);
        else if (are_connected({b.back()}, {i})) b.push_back(i);
        else {
          reverse(b.begin(), b.end());
          for(int &j : b) a.push_back(j);
          b = {i};
        }
      }
      if (a.size() > b.size()) swap(a, b);

      if (a.size() > 2 && !are_connected({a[0]}, {a.back()})) {
        if (are_connected({a[0]}, {b[0]})) return sum(rev(a), b);
        else return sum(a, b);
      }
      if (b.size() > 2 && !are_connected({b[0]}, {b.back()})) {
        if (are_connected({a[0]}, {b[0]})) return sum(rev(a), b);
        else return sum(b, a);
      }

      auto [x, y] = Find(a, b);
      if (x < 0) return b;
      vector<int> ans;
      for(int j = x + 1; j < a.size(); j++) ans.push_back(a[j]);
      for(int j = 0; j <= x; j++) ans.push_back(a[j]);
      for(int j = y; j < b.size(); j++) ans.push_back(b[j]);
      for(int j = 0; j < y; j++) ans.push_back(b[j]);
      return ans;
    }
}

Compilation message

longesttrip.cpp: In function 'std::pair<int, int> Find(std::vector<int>, std::vector<int>)':
longesttrip.cpp:12:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   12 |     int mid = l + r >> 1;
      |               ~~^~~
longesttrip.cpp:22:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   22 |     int mid = l + r >> 1;
      |               ~~^~~
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:80:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   80 |       for(int j = x + 1; j < a.size(); j++) ans.push_back(a[j]);
      |                          ~~^~~~~~~~~~
longesttrip.cpp:82:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   82 |       for(int j = y; j < b.size(); j++) ans.push_back(b[j]);
      |                      ~~^~~~~~~~~~
longesttrip.cpp:86:1: warning: control reaches end of non-void function [-Wreturn-type]
   86 | }
      | ^
# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 2 ms 856 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 11 ms 344 KB Output is correct
2 Correct 9 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 5 ms 344 KB Output is correct
5 Correct 4 ms 596 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 9 ms 344 KB Output is correct
2 Correct 7 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 5 ms 344 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Correct 11 ms 344 KB Output is correct
7 Correct 8 ms 344 KB Output is correct
8 Correct 7 ms 344 KB Output is correct
9 Correct 5 ms 340 KB Output is correct
10 Correct 5 ms 436 KB Output is correct
11 Correct 5 ms 440 KB Output is correct
12 Correct 5 ms 440 KB Output is correct
13 Correct 5 ms 640 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 12 ms 344 KB Output is correct
2 Correct 8 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 5 ms 600 KB Output is correct
5 Correct 5 ms 344 KB Output is correct
6 Correct 12 ms 344 KB Output is correct
7 Correct 7 ms 344 KB Output is correct
8 Correct 5 ms 344 KB Output is correct
9 Correct 6 ms 600 KB Output is correct
10 Correct 5 ms 440 KB Output is correct
11 Correct 5 ms 440 KB Output is correct
12 Correct 5 ms 612 KB Output is correct
13 Correct 5 ms 440 KB Output is correct
14 Correct 7 ms 344 KB Output is correct
15 Correct 7 ms 344 KB Output is correct
16 Correct 8 ms 344 KB Output is correct
17 Correct 6 ms 600 KB Output is correct
18 Correct 6 ms 344 KB Output is correct
19 Correct 6 ms 608 KB Output is correct
20 Correct 6 ms 600 KB Output is correct
21 Correct 5 ms 608 KB Output is correct
22 Correct 5 ms 700 KB Output is correct
23 Correct 5 ms 692 KB Output is correct
24 Correct 6 ms 952 KB Output is correct
25 Correct 8 ms 344 KB Output is correct
26 Correct 7 ms 344 KB Output is correct
27 Correct 6 ms 344 KB Output is correct
28 Correct 6 ms 344 KB Output is correct
29 Correct 7 ms 344 KB Output is correct
30 Correct 4 ms 344 KB Output is correct
31 Correct 5 ms 344 KB Output is correct
32 Correct 6 ms 344 KB Output is correct
33 Correct 5 ms 600 KB Output is correct
34 Correct 5 ms 436 KB Output is correct
35 Correct 6 ms 344 KB Output is correct
36 Correct 4 ms 344 KB Output is correct
37 Correct 6 ms 344 KB Output is correct
38 Correct 5 ms 608 KB Output is correct
39 Correct 5 ms 344 KB Output is correct
40 Correct 6 ms 600 KB Output is correct
41 Correct 7 ms 600 KB Output is correct
42 Correct 8 ms 600 KB Output is correct
43 Correct 8 ms 600 KB Output is correct
44 Correct 7 ms 344 KB Output is correct
45 Correct 12 ms 348 KB Output is correct
46 Correct 9 ms 344 KB Output is correct
47 Correct 8 ms 344 KB Output is correct
48 Correct 7 ms 344 KB Output is correct
49 Correct 7 ms 344 KB Output is correct
50 Correct 6 ms 588 KB Output is correct
51 Correct 8 ms 632 KB Output is correct
52 Correct 8 ms 340 KB Output is correct
53 Correct 5 ms 600 KB Output is correct
54 Correct 6 ms 608 KB Output is correct
55 Correct 6 ms 600 KB Output is correct
56 Correct 5 ms 444 KB Output is correct
57 Correct 7 ms 440 KB Output is correct
58 Correct 9 ms 696 KB Output is correct
59 Correct 8 ms 604 KB Output is correct
60 Correct 10 ms 604 KB Output is correct
61 Correct 9 ms 856 KB Output is correct
62 Correct 9 ms 440 KB Output is correct
63 Correct 8 ms 604 KB Output is correct
64 Correct 8 ms 600 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 13 ms 344 KB Output is correct
2 Correct 8 ms 344 KB Output is correct
3 Correct 6 ms 344 KB Output is correct
4 Correct 5 ms 344 KB Output is correct
5 Correct 4 ms 344 KB Output is correct
6 Correct 10 ms 344 KB Output is correct
7 Correct 7 ms 344 KB Output is correct
8 Correct 5 ms 344 KB Output is correct
9 Correct 5 ms 344 KB Output is correct
10 Correct 5 ms 604 KB Output is correct
11 Correct 7 ms 436 KB Output is correct
12 Correct 6 ms 596 KB Output is correct
13 Correct 6 ms 856 KB Output is correct
14 Correct 6 ms 344 KB Output is correct
15 Correct 8 ms 344 KB Output is correct
16 Correct 11 ms 344 KB Output is correct
17 Correct 6 ms 600 KB Output is correct
18 Correct 6 ms 344 KB Output is correct
19 Correct 6 ms 344 KB Output is correct
20 Correct 6 ms 600 KB Output is correct
21 Correct 8 ms 344 KB Output is correct
22 Correct 10 ms 344 KB Output is correct
23 Correct 6 ms 344 KB Output is correct
24 Correct 6 ms 344 KB Output is correct
25 Correct 7 ms 580 KB Output is correct
26 Correct 7 ms 344 KB Output is correct
27 Correct 6 ms 344 KB Output is correct
28 Correct 6 ms 344 KB Output is correct
29 Correct 5 ms 600 KB Output is correct
30 Correct 5 ms 600 KB Output is correct
31 Correct 5 ms 344 KB Output is correct
32 Correct 10 ms 344 KB Output is correct
33 Correct 8 ms 344 KB Output is correct
34 Correct 8 ms 344 KB Output is correct
35 Correct 8 ms 344 KB Output is correct
36 Correct 10 ms 344 KB Output is correct
37 Correct 5 ms 436 KB Output is correct
38 Correct 8 ms 436 KB Output is correct
39 Correct 9 ms 344 KB Output is correct
40 Correct 6 ms 600 KB Output is correct
41 Correct 6 ms 344 KB Output is correct
42 Correct 5 ms 344 KB Output is correct
43 Correct 8 ms 696 KB Output is correct
44 Correct 6 ms 952 KB Output is correct
45 Correct 6 ms 696 KB Output is correct
46 Correct 5 ms 604 KB Output is correct
47 Correct 5 ms 608 KB Output is correct
48 Correct 5 ms 596 KB Output is correct
49 Correct 5 ms 608 KB Output is correct
50 Correct 7 ms 344 KB Output is correct
51 Partially correct 7 ms 344 KB Output is partially correct
52 Correct 6 ms 600 KB Output is correct
53 Correct 7 ms 344 KB Output is correct
54 Correct 8 ms 344 KB Output is correct
55 Correct 7 ms 344 KB Output is correct
56 Correct 5 ms 344 KB Output is correct
57 Partially correct 6 ms 344 KB Output is partially correct
58 Partially correct 9 ms 344 KB Output is partially correct
59 Partially correct 7 ms 344 KB Output is partially correct
60 Correct 7 ms 344 KB Output is correct
61 Correct 6 ms 344 KB Output is correct
62 Correct 7 ms 608 KB Output is correct
63 Partially correct 8 ms 444 KB Output is partially correct
64 Partially correct 9 ms 604 KB Output is partially correct
65 Partially correct 7 ms 608 KB Output is partially correct
66 Partially correct 8 ms 696 KB Output is partially correct
67 Correct 8 ms 436 KB Output is correct
68 Correct 7 ms 692 KB Output is correct
69 Partially correct 8 ms 692 KB Output is partially correct
70 Partially correct 8 ms 856 KB Output is partially correct
71 Correct 6 ms 856 KB Output is correct
72 Partially correct 7 ms 800 KB Output is partially correct
73 Partially correct 9 ms 604 KB Output is partially correct
74 Partially correct 7 ms 604 KB Output is partially correct
75 Correct 7 ms 436 KB Output is correct
76 Correct 8 ms 608 KB Output is correct
77 Correct 7 ms 444 KB Output is correct
78 Partially correct 7 ms 444 KB Output is partially correct
79 Correct 7 ms 604 KB Output is correct
80 Partially correct 7 ms 696 KB Output is partially correct
81 Correct 7 ms 696 KB Output is correct
82 Correct 7 ms 600 KB Output is correct