HDU 5289 Assignment

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);
    }
}

猜你喜欢

转载自blog.csdn.net/lml11111/article/details/81071367