| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 1361477 | jalol250 | Triple Peaks (IOI25_triples) | C++20 | 0 ms | 0 KiB |
#include <bits/stdc++.h>
using namespace std;
#define int long long
long long count_triples(vector<int> H){
int n = H.size();
long long ans = 0;
for(int i=0;i<n;i++){
for(int j=i+1;j<n;j++){
for(int k=j+1;k<n;k++){
vector<int> a = {H[i],H[j],H[k]};
vector<int> b = {j-i, k-i, k-j};
sort(a.begin(),a.end());
sort(b.begin(),b.end());
if(a==b) ans++;
}
}
}
return ans;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int n;
cin>>n;
vector<int> v;
for(int i=0;i<n;i++){
cin>>v[i];
}
cout<<count_triples(v);
}