| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1282622 | kawhiet | 3개의 봉우리 (IOI25_triples) | C++20 | 0 ms | 0 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;
}