51nod1275 连续子段的差异(单调队列)

给出一个包括N个元素的整数数组A,包括A本身在内,共有 (N+1)*N / 2个非空子段。例如:1 3 2的子段为{1} {3} {2} {1 3} {3 2} {1 3 2}。在这些子段中,如果最大值同最小值的差异不超过K,则认为这是一个合格的子段。给出数组A和K,求有多少符合条件的子段。例如:3 5 7 6 3,K = 2,符合条件的子段包括:{3} {5} {7} {6} {3} {3 5} {5 7} {7 6} {5 7 6},共9个。
Input
第1行:2个数N, K(1 <= N <= 50000, 0 <= K <= 10^9)
第2 - N + 1行:每行1个数,对应数组的元素Ai(0 <= A[i] <= 10^9)
Output
输出符合条件的子段数量。

可以看出,如果一个数列满足要求,那么这个数列的每个连续子集肯定满足要求,求任意一段数列的最值可以用单调队列完成,一段左端点为L,右端点为R的数列全部连续子序列数量应该是(r-l+2)*(r-l+1)/2,所以我一开始的想法就是每次到一个符合题意的数列,就用这种办法统计方案数,但是这种统计方法是有重复的。

如这组测试数据:5 3
1 2 3 4 5
1-4和2-5均为符合题意的数列,如果按照上面的方法统计就会重复,比如2-4数列就被算了两次,所以是错误的。

正确的办法应该是枚举左端点,如果当前数列左为L右为R,那么(L,L+1),…
(L,R),这些区间不会和前面有重复,因为枚举左端点,如果L-1也按这种办法枚举,这两种区间是不会有交集的,所以这种统计方法保证了不会重复。

最后,如果对于L其最远符合要求位置为R,那么对于L+1,其满足要求的最远右端点不可能比R小,所以每个点最后只会出队入队一次,复杂度为O(n).

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define maxn 50005
#define angel 0x3f3f3f3f
int num[maxn],deq_max[maxn],deq_min[maxn];//分别为维护最大值队列和最小值队列
int main()
{
    ll n,k;
    scanf("%lld%lld",&n,&k);
    for(int i=1; i<=n; i++)
        scanf("%d",&num[i]);
    int s_max=0,s_min=0,t_max=0,t_min=0;
    ll ans=0;
    ll l=1,r=1;
    while(l<=n)
    {
        while(s_max<t_max&&num[deq_max[t_max-1]]<=num[r])
            t_max--;
        deq_max[t_max++]=r;
        while(s_min<t_min&&num[deq_min[t_min-1]]>=num[r])
            t_min--;
        deq_min[t_min++]=r;
        if(num[deq_max[s_max]]-num[deq_min[s_min]]<=k&&r<n)
            r++;
        else
        {
            if(num[deq_max[s_max]]-num[deq_min[s_min]]<=k)
                ans=ans+r-l+1;
            else
                ans=ans+r-l;
            if(deq_max[s_max]<=l)
                s_max++;
            if(deq_min[s_min]<=l)
                s_min++;
            l++;
        }
    }
    printf("%lld\n",ans);
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值