Gym - 103185E E - Excellent Views

题目大意:
给你个n与n个数字的数组
对于每个数字的下标设为i,值设为Hi
看有多少下标为j的数字没有k符合|i−k|≤|i−j| 并且 Hj

思路:
不等式就是表示距离嘛,公式的意思就是一个点向两边散开递增的数字有多少个
那这个意思其实跟单调栈还挺像的
不过呢单调栈都是单方向的向左或者向右一个方向的
这个则是要双向并且递增的
看了师兄的代码
这里的做法是看每个点对左边的点和对右边的点的贡献
首先如果是从左往右扫,对于a[i]
用一个单调栈,遇到下一个a[j] 就说明a[i]对(i+j)/2前的元素是有+1的贡献的(因为这时候距离近嘛)
然后优化一下用一个差分记录一下+1的贡献
然后再从有往左扫,跟上面同理代码就出来了

AC代码:

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define pf(x) (x)*(x)
#define lowerbit(x) x&(-x)
#define mid (l+r>>1)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll N=1e5+10;
const int inf=0x3f3f3f3f;
int a[N];
int st[N];
int dp[N];
int sum[N];
int to[N];

int main()
{
    ios::sync_with_stdio(false),cin.tie(0);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    int top=0;
    st[++top]=1;
    a[n+1]=inf;
    for(int i=2;i<=n+1;i++){//考虑每个a[i]对左边的贡献
        while(top>0&&a[i]>a[st[top]]){
            to[st[top--]]=i-1;
        }
        st[++top]=i;
    }
    for(int i=1;i<=n;i++){//影响范围是中点前的那一段,对这些点贡献+1
        dp[i+1]++;
        if(to[i]!=n)dp[(i+to[i])/2+1]--;//
    }
    for(int i=1;i<=n;i++){
        dp[i]+=dp[i-1];//差分
    }
    top=0;
    a[0]=inf;
    st[++top]=n;
    for(int i=n-1;i>=0;i--){//考虑每个a[i]对右边的贡献
        while(top>0&&a[i]>a[st[top]])to[st[top--]]=i+1;
        st[++top]=i;
    }
    for(int i=1;i<=n;i++){//跟上面同理
        sum[i-1]++;
        if(to[i]!=1)sum[(i+to[i]+1)/2-1]--;
    }
    for(int i=n;i>=1;i--)sum[i]+=sum[i+1];
    for(int i=1;i<=n;i++)cout<<dp[i]+sum[i]<<' ';
    return 0;
}

你可能感兴趣的:(题解,算法)