ABC140 E - Second Sum（排序+multiset）

题意：

数据范围：n<=1e5

解法：

第一想法是计算每个数能在多少区间内能作为第二大的数.
枚举这个数,要计算满足条件的区间,
就需要知道这个数左边和右边第一个和第二个比他大的数的位置.
一种做法是将所有数存为(val,pos),按val从大到小排序,
用一个set插入大于当前数的所有数的位置,
然后就可以二分找到左边和右边第一个和第二个比他大的数的位置了.
利用了值的单调性(排序之后值递减).

参考:
https://www.cnblogs.com/heyuhhh/p/11491647.html

图示：
其中ll，l，r，rr是now左边和右边第一个和第二个大于now的位置。
第一种情况：

第二种情况：

code：

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define PI pair<int,int>
const int maxm=1e6+5;
struct Node{
    int v,id;
}e[maxm];
int a[maxm];
int n;
bool cmp(Node a,Node b){
    return a.v>b.v;
}
signed main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        e[i]={a[i],i};
    }
    sort(e+1,e+1+n,cmp);
    multiset<int>s;
    s.insert(0);//插入边界,这样的话之后就不用特判了.
    s.insert(0);
    s.insert(n+1);
    s.insert(n+1);
    int ans=0;
    for(int i=1;i<=n;i++){
        int x=e[i].v,p=e[i].id;
        s.insert(p);
        auto it=s.lower_bound(p);//==p
        auto r=it;r++;
        auto rr=r;rr++;
        auto l=it;l--;
        auto ll=l;ll--;
        if(*l>=1)ans+=(*l-*ll)*(*r-p)*x;
        if(*r<=n)ans+=(*rr-*r)*(p-*l)*x;
    }
    cout<<ans<<endl;
    return 0;
}

---