Submission #1282622

#TimeUsernameProblemLanguageResultExecution timeMemory
1282622kawhietTriple Peaks (IOI25_triples)C++20
Compilation error
0 ms0 KiB
#include <bits/stdc++.h>
#include "triples.h"
using namespace std;

long long count_triples(vector<int> h) {
  int n = h.size();
  set<array<int, 3>> s;
  auto f = [&](int i, int j, int k) {
    array<int, 3> c = {i, j, k};
    sort(c.begin(), c.end());
    if (s.count(c)) {
      return false;
    }
    vector<int> a = {abs(j - i), abs(k - i), abs(k - j)};
    vector<int> b = {h[i], h[j], h[k]};
    sort(a.begin(), a.end());
    sort(b.begin(), b.end());
    if (a == b) {
      s.insert(c);
      return true;
    }
    return false;
  };
  long long res = 0;
  for (int i = 0; i < n; i++) {
    if (i >= h[i]) {
      int j = i - h[i], k = j + h[j];
      res += (k < i && f(i, j, k));
      k = i - h[j];
      res += (k > j && f(i, j, k));
    }
    if (i + h[i] < n) {
      int j = i + h[i], k = j - h[j];
      res += (k > i && f(i, j, k));
      k = i + h[j];
      res += (k < j && f(i, j, k));
    }
  }
  return res;
}

Compilation message (stderr)

/usr/bin/ld: /tmp/cc0Zdg7S.o: in function `main':
grader.cpp:(.text.startup+0x197): undefined reference to `construct_range(int, int)'
collect2: error: ld returned 1 exit status