解题思路:对于每一个点,去往前找比它大的第一个数,去往后找比他大的第一个数,然后算min(arr[left]-arr[i],arr[right]-arr[i]),答案加上这个值乘以(right-left-1),如果在往前找的时候,发现某个数和他相等,那么说明之前已经处理过了,就不处理了,直接返回。
#include<bits/stdc++.h>
#define MAX 10001
using namespace std;
int n;
int arr[MAX];
long long res=0;
int left_high(int j)
{
for(int i=j-1;i>=0;i--)
{
if(arr[i]==arr[j])
{
return -1;
}
if(arr[i]>arr[j])
{
return i;
}
}
return -1;
}
int right_hight(int j)
{
for(int i=j+1;i<n;i++)
{
if(arr[i]>arr[j])
{
return i;
}
}
return -1;
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
{
cin>>arr[i];
}
int left=-1,right=-1,temp=0;
for(int i=1;i<n;i++)
{
if(arr[i]==arr[i-1]) continue;
left = left_high(i);
if(left !=-1)
{
right = right_hight(i);
if(right!=-1)
{
temp = min(arr[left]-arr[i],arr[right]-arr[i]);
//cout<<left<<" "<<right<<" "<<temp<<endl;
res+=(temp*(right-left-1));
}
}
}
cout<<res<<endl;
return 0;
}