| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1282639 | 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(i - j), abs(i - k), abs(j - k)};
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++) {
for (int j = 0; j + h[i] < n; j++) {
int k = j + h[i];
if (j != i && k != i) {
res += f(i, j, k);
}
}
}
return res;
}
vector<int> construct_range(int M, int K) {
return {1, 1, 1};
}
int main() {
int n;
cin >> n;
vector<int> h(n);
for (int i = 0; i < n; i++) {
cin >> h[i];
}
cout << count_triples(h) << '\n';
}
