题意
思路
我们考虑对于每一个数,有多少个区间的第二大数为该数,即可得到该数的贡献;
若当前数 x 位于下标 i 的位置,可向左右找第一个大于该数的数的位置,记为 l,r,继续找第二个大于该数的数的位置,记为 ll,rr,则该数的贡献即为,表示 x 作为第二大数,第一大数分别为右边第一个比 x 的数,或左边第一个比 x 的数;
现在问题转换为如何快速找到,两边比当前数大的第一,第二大的数,如果我们从大到小依次将数,填入数组,则当填入 x 时,在数组中的数一定比大,此时直接找 x 位置的前驱后继即可,也就是平衡树,可以用 set 维护,复杂度 ;
注意处理边界;
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[100010],b[100010];
set<int> st;
int main()
{
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
int n;
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i]; b[a[i]]=i;
}
st.insert(0); st.insert(n+1); st.insert(b[n]);
ll ans=0;
for(ll i=n-1;i>=1;i--){
st.insert(b[i]);
auto itr=st.lower_bound(b[i]);
auto itr1=itr,itr2=itr;
itr1--; itr2++;
ll l=*itr1,r=*itr2,ll,rr;
if(l){
itr1--; ll=*itr1;
}
else ll=0;
if(r<n+1){
itr2++; rr=*itr2;
}
else rr=n+1;
ans+=i*(rr-r)*(b[i]-l)+i*(l-ll)*(r-b[i]);
}
cout<<ans<<"\n";
return 0;
}