#include <iostream>
#include <algorithm>
using namespace std;
int Lcounts[100002], Rcounts[100002];
long long zck(int A[], int n)
{
long long ans = 0;
if (n == 1)
return ans;
int mid = n / 2, i = mid-1, j = mid;
int Allmin = min(A[i], A[j]);
Lcounts[A[i]] = 1;
Rcounts[A[j]] = 1;
if (abs(A[i] - A[j]) == Allmin)
ans = 1;
for(int k = 0; k < n-2; k++){
if (i == 0){
Allmin = min(Allmin, A[++j]);
Rcounts[A[j]]++;
ans += Lcounts[min(A[j] + Allmin, 100001)];
ans += Lcounts[A[j] - Allmin];
}else if (j == n-1){
Allmin = min(Allmin, A[--i]);
Lcounts[A[i]]++;
ans += Rcounts[min(A[i] + Allmin, 100001)];
ans += Rcounts[A[i] - Allmin];
}else{
if (min(A[i-1],Allmin) > min(A[j+1],Allmin)){
Allmin = min(Allmin, A[--i]);
Lcounts[A[i]]++;
ans += Rcounts[min(A[i] + Allmin, 100001)];
ans += Rcounts[A[i] - Allmin];
}else{
Allmin = min(Allmin, A[++j]);
Rcounts[A[j]]++;
ans += Lcounts[min(A[j] + Allmin, 100001)];
ans += Lcounts[A[j] - Allmin];
}
}
}
for (int k = 0; k < mid; k++){
Lcounts[A[k]] = 0;
}
for (int k = mid; k < n; k++){
Rcounts[A[k]] = 0;
}
ans += (zck(A, mid) + zck(A + mid, n - mid));
return ans;
}
void Read(int & p)
{
p=0;
int flag=1;
char c=getchar();
while(c<'0' or c>'9')
{
if(c=='-') flag=-1;
c=getchar();
}
while(c>='0' and c<='9')
p=p*10+(c-'0'),c=getchar();
p*=flag;
}
int main()
{
int N;
Read(N);
int A[N];
for (int i = 0; i < N; i++){
Read(A[i]);
}
cout << zck(A,N);
}
Write, Run & Share C++ code online using OneCompiler's C++ online compiler for free. It's one of the robust, feature-rich online compilers for C++ language, running on the latest version 17. Getting started with the OneCompiler's C++ compiler is simple and pretty fast. The editor shows sample boilerplate code when you choose language as C++ and start coding!
OneCompiler's C++ online compiler supports stdin and users can give inputs to programs using the STDIN textbox under the I/O tab. Following is a sample program which takes name as input and print your name with hello.
#include <iostream>
#include <string>
using namespace std;
int main()
{
string name;
cout << "Enter name:";
getline (cin, name);
cout << "Hello " << name;
return 0;
}
C++ is a widely used middle-level programming language.
When ever you want to perform a set of operations based on a condition If-Else is used.
if(conditional-expression) {
//code
}
else {
//code
}
You can also use if-else for nested Ifs and If-Else-If ladder when multiple conditions are to be performed on a single variable.
Switch is an alternative to If-Else-If ladder.
switch(conditional-expression){
case value1:
// code
break; // optional
case value2:
// code
break; // optional
......
default:
code to be executed when all the above cases are not matched;
}
For loop is used to iterate a set of statements based on a condition.
for(Initialization; Condition; Increment/decrement){
//code
}
While is also used to iterate a set of statements based on a condition. Usually while is preferred when number of iterations are not known in advance.
while (condition) {
// code
}
Do-while is also used to iterate a set of statements based on a condition. It is mostly used when you need to execute the statements atleast once.
do {
// code
} while (condition);
Function is a sub-routine which contains set of statements. Usually functions are written when multiple calls are required to same set of statements which increases re-usuability and modularity. Function gets run only when it is called.
return_type function_name(parameters);
function_name (parameters)
return_type function_name(parameters) {
// code
}