//#include "triplepeaks.h"
#include <bits/stdc++.h>
#define task "TEST"
#define task2 "A"
#define pl pair<ll, ll>
#define pf push_front
#define pb push_back
#define pob pop_back
#define pof pop_front
#define mp make_pair
#define fi first
#define se second
#define FOR(i, a, b, c) for (int i=a; i<=b; i+=c)
#define FORE(i, a, b, c) for (int i=a; i>=b; i+=c)
using namespace std;
using ll = long long;
using ull = unsigned long long;
const int Mod = 998244353;
const int maxn = 1e3;
const ll Inf = 1e16;
ll n;
ll Check(ll i, ll j, ll k, vector<int>& H) {
if (j < i or k < j) return 0;
if (i < 0 or k >= n) return 0;
multiset<int> a, b;
a.insert({k-i, j-i, k-j});
b.insert({H[i], H[j], H[k]});
if (a == b) cerr << i << " " << j << " " << k << '\n';
return (a == b);
}
ll doubles(ll i, ll j, vector<int>& H) {
ll v = abs(j-i);
ll a = min(H[i], H[j]), b = max(H[i], H[j]);
if (v != H[i] and v != H[j]) {
ll v1 = Check(i, j, j + a, H), v2 = Check(i, j, j + b, H);
if (a == b) return v1;
else return v1 + v2;
}
else return Check(i, j, j+v, H);
}
ll count_triples(vector<int> H) {
n = H.size(); ll cnt = 0;
FOR(i, 0, n-3, 1) FOR(j, i+1, n-2, 1)
cnt += doubles(i, j, H);
return cnt;
}
vector<int> construct_range(int M, int K) {
vector<int> a; a.clear();
return a;
}
/*int main() {
count_triples({4, 1, 4, 3, 2, 6, 1});
}*/
Compilation message (stderr)
triples.cpp: In function 'll Check(ll, ll, ll, std::vector<int>&)':
triples.cpp:26:16: warning: narrowing conversion of '(k - i)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
26 | a.insert({k-i, j-i, k-j});
| ~^~
triples.cpp:26:16: warning: narrowing conversion of '(k - i)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
triples.cpp:26:21: warning: narrowing conversion of '(j - i)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
26 | a.insert({k-i, j-i, k-j});
| ~^~
triples.cpp:26:21: warning: narrowing conversion of '(j - i)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
triples.cpp:26:26: warning: narrowing conversion of '(k - j)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
26 | a.insert({k-i, j-i, k-j});
| ~^~
triples.cpp:26:26: warning: narrowing conversion of '(k - j)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |