代码源每日一题Div.1 第二大数字和（set）

题目链接

题意

思路

     我们考虑对于每一个数，有多少个区间的第二大数为该数，即可得到该数的贡献；

     若当前数 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;
}