[优先队列]HDOJ5289 Assignment

题意：有多少个区间，区间内最大的数减去最小的数差小于k

对每个数它所在的区间，可以只往前找(类似dp的无后效性) 比如对位置3的数，可以往前找的区间是[3, 3], [2, 3], [1, 3], [0, 3]这样

这样 遍历a[i] 对每个a[i]往前 就可以得到所有的区间

比如案例

10 5

0 3 4 5 2 1 6 7 8 9

(最小值，最大值)的形式

每个数增加一个(自己，自己)

一旦不满足条件了就不必再进行下去

那么很容易想到这是一个类似队列的情况

对于这个案例 跑一下队列就是

 (把队中元素的个数加起来就是结果)

那么问题就转化成了怎么知道队列中存在一个数是要弹出的(也就是上述案例6的时候，怎么知道1在队列中)

我先用优先队列处理了一下每个数前面需要弹出的数的位置的最大值

再用队列模拟上述过程即可

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
#define lson l, m
#define rson m+1, r
const int N=1e5+5;
const LL mod=1e9+7;
typedef pair<int, int> PI;
PI a[N];
int minn[N], maxn[N];
void out(priority_queue<PI, vector<PI>, greater<PI> > q)
{
    while(!q.empty())
    {
        printf("(%d %d)  ", q.top().first, q.top().second);
        q.pop();
    }
    puts("");
}

int main()
{
//    freopen("1002.in", "r", stdin);
//    freopen("out.txt", "w", stdout);
    int t;
    scanf("%d", &t);
    while(t--)
    {
        int n, k;
        scanf("%d%d", &n, &k);
        for(int i=0;i<n;i++)
        {
            int x;
            scanf("%d", &x);
            a[i]=make_pair(x, i);
        }
        memset(minn, -1, sizeof(minn));
        memset(maxn, -1, sizeof(maxn));
        priority_queue<PI, vector<PI>, greater<PI> > q1;
        while(!q1.empty())
            q1.pop();
        q1.push(a[0]);
        for(int i=1;i<n;i++)
        {
            while(q1.size()>0 && a[i].first-q1.top().first>=k)
            {
                minn[i]=max(q1.top().second, minn[i]);
                q1.pop();
            }
            q1.push(a[i]);
//            out(q1);
        }
        priority_queue<PI, vector<PI>, less<PI> > q2;
        while(!q2.empty())
            q2.pop();
        q2.push(a[0]);
        for(int i=1;i<n;i++)
        {
            while(q2.size()>0 && q2.top().first-a[i].first>=k)
            {
                maxn[i]=max(q2.top().second, maxn[i]);
                q2.pop();
            }
            q2.push(a[i]);
        }
//        for(int i=0;i<n;i++)
//            printf("%d %d
", minn[i], maxn[i]);

        LL ans=0;
        queue<int> Q;
        while(!Q.empty())
            Q.pop();
        int cur=0;
        for(int i=0;i<n;i++)
        {
            if(minn[i]<cur)
                minn[i]=-1;
            if(maxn[i]<cur)
                maxn[i]=-1;
            while(!Q.empty() && (maxn[i]!=-1 || minn[i]!=-1))
            {
                if(Q.front()==maxn[i])
                    maxn[i]=-1;
                if(Q.front()==minn[i])
                    minn[i]=-1;
                cur=Q.front();
                Q.pop();
            }
            Q.push(a[i].second);
            ans+=Q.size();
        }
        printf("%I64d
", ans);
    }
    return 0;
}