HDU 5289 Assignment

最新推荐文章于 2022-01-02 19:02:52 发布

尒轩窗

最新推荐文章于 2022-01-02 19:02:52 发布

阅读量196

点赞数

分类专栏：单调队列

本文链接：https://blog.csdn.net/lml11111/article/details/81071367

版权

单调队列专栏收录该内容

4 篇文章 0 订阅

订阅专栏

http://acm.hdu.edu.cn/showproblem.php?pid=5289

题意：

分组方式是连续的数中最大值和最小值的差小于K，有几种分组方式（不一定用n个数）

分析：

用双向队列进行两边的删除、插入操作，左边的删除操作是不满足小于K的条件就把左边界右移直到满足条件，右边的操作是当是维护最大值是，保证输入的数比前面的数都小，如果前面的数比他小就把他弹出

代码：

#include<bits/stdc++.h>
using namespace std;
int a[100005];
#define mod 1000000007
deque<int>high,low;
int main()
{
    int i,j,l,r,t,n,k;
    long long ans;
    scanf("%d",&t);
    while(t--)
    {
        high.clear();
        low.clear();
        scanf("%d%d",&n,&k);
        ans=0;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(i=j=1;i<=n;i++)
        {
            while(!high.empty()&&high.back()<a[i])high.pop_back();
            high.push_back(a[i]);
            while(!low.empty()&&low.back()>a[i])low.pop_back();
            low.push_back(a[i]);
            while(!high.empty()&&!low.empty()&&high.front()-low.front()>=k)
            {
                ans+=i-j;
                if(high.front()==a[j])high.pop_front();
                if(low.front()==a[j])low.pop_front();
                j++;
            }
        }
        while(j<=n)
        {
            ans+=i-j;
            j++;
        }
        printf("%lld\n",ans);
    }
}