2021年广东工业大学第十五届文远知行杯程序设计竞赛 B.找山坡（单调栈+set）

题意：

解法：

单调栈计算出每个元素作为最小值可以扩展的最右位置r[]

题目要求a[L]=a[R],
如果区间[L,R]满足条件,那么r[L]>=R一定成立.

因此对于每个a[R],我们只需要找到a[L]=a[R]且r[L]>=R的L,用R-L更新答案.
我的做法是对于每个x,将a[i]=x的i全部丢入同一个vector,
在这个vector种,对于每个i,我们要找它左边的j,满足r[j]>=i,如果有多个满足,找最小的j.

开一个set存左边所有满足r[j]>=i的所有j,设这个set为S1.
再开一个set存(r[j],j)二元组,设这个set为S2.

观察到在i右移的过程中,如果r[j]<i,那么r[j]也<i+1,那么将j给去掉,
具体做法是将S2中,(r[j],j)中r[j]<i的二元组取出,将其对应的j从S1中删除.

之后S1中的j一定都是满足条件的,我们取最小的j出来,用i-j更新答案即可.

算法总复杂度O(n*log)

code：

#include<bits/stdc++.h>
#define int long long
#define PI pair<int,int>
using namespace std;
const int maxm=1e6+5;
vector<int>g[maxm];
int stk[maxm];
int r[maxm];
int a[maxm];
int b[maxm];
int n;
int cal(int x){
    int len=g[x].size();
    if(len<=1)return 0;
    int ans=0;
    set<int>idx;
    set<PI>R;
    for(int i=0;i<len;i++){
        int pos=g[x][i];
        while(R.size()){
            auto it=R.begin();
            if(it->first<pos){
                idx.erase(it->second);
                R.erase(it);
            }else{
                break;
            }
        }
        if(idx.size()){
            ans=max(ans,pos-*idx.begin());
        }
        R.insert({r[pos],pos});
        idx.insert(pos);
    }
    return ans;
}
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    //离散化
    for(int i=1;i<=n;i++){
        b[i]=a[i];
    }
    sort(b+1,b+1+n);
    int num=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=n;i++){
        a[i]=lower_bound(b+1,b+1+num,a[i])-b;
    }
    //
    a[n+1]=-1;
    //单调栈计算出每个数作为最小值能扩展的最右位置
    int head=0;
    for(int i=1;i<=n+1;i++){
        while(head&&a[i]<a[stk[head]]){
            r[stk[head]]=i-1;
            head--;
        }
        stk[++head]=i;
    }
    //
    for(int i=1;i<=n;i++){
        g[a[i]].push_back(i);
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        ans=max(ans,cal(i));
    }
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---