| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1288016 | nikoloz-ch | 3개의 봉우리 (IOI25_triples) | C++20 | 0 ms | 0 KiB |
#include <bits/stdc++.h>
using namespace std;
long long count_triples(std::vector<int> H){
int N = H.size(), ans = 0;
if(N <= 5000){
map<int,vector<int>> mp;
for(int i = 0; i < N; i++){
mp[i-H[i]].push_back(i);
}
for(int i = 0; i < N; i++){
for(int j = i+1; j < N; j++){
const int k = j-i;
if(k != H[i] and k != H[j]){
if(k != abs(H[i]-H[j])) continue;
const int p = j+min(H[i],H[j]);
if(p >= N or H[p] != k or p <= j) continue;
ans++;
continue;
}
if(k == H[i]){
for(auto &t : mp[i]){
if(t <= j) continue;
if(t-j != H[j]) continue;
ans++;
}
for(auto &t : mp[j]){
if(t <= j) continue;
if(t-i != H[j]) continue;
ans++;
}
continue;
}
for(auto &t : mp[i]){
if(t <= j) continue;
if(abs(t-j) != H[i]) continue;
ans++;
}
for(auto &t : mp[j]){
if(t <= j) continue;
if(abs(t-i) != H[i]) continue;
ans++;
}
continue;
}
}
return ans;
}
}
