HDU 5806 (dp)

问题描述:
在数学课上，NanoApe 心痒痒又玩起了数列。他在纸上随便写了一个长度为 $n$ 的数列，他又根据心情写下了一个数 $m$。 他想知道这个数列中有多少个区间里的第 $k$大的数不小于 $m$，当然首先这个区间必须至少要有 $k$ 个数啦。

输入描述
第一行为一个正整数 $T$，表示数据组数。
每组数据的第一行为三个整数 $n$ , $m$ , $k$
第二行为 $n$ 个整数$Ai$，表示这个数列。
$1 < =T <=10$ , $2 <= n <= 200000$ , $1 <= k <= n/2$ , $1 <= m$ , $Ai <=10^9$

输出描述:
对于每组数据输出一行一个数表示答案。

解题思路：
这题一看就会让人想到$dp$，$dp[i]$ 表示到 $i$ 位置之前的满足条件的序列个数，下标从$0$开始，用一个队列 $q$ 将$i$ 位置之前大于&m&的下标进队。当扫描到的当前位置$i$的值：
小于m时$dp[i]=dp[i-1]+q.front()+1$
当大于等于 $m$ 时,先让当前下标值进队，队首$pop()$,因为对中只需保留k个下标即可。 $dp[i]=dp[i-1]+q.front()+1$。具体看代码。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>

using namespace std;

#define ll long long

queue <ll>q;

ll a[200050];
ll dp[200050];
int main(){
    ll T,num,tmp,ans,n,m,k;
    scanf("%I64d",&T);
    while(T--)
    {
        while(!q.empty())
            q.pop();
        scanf("%I64d%I64d%I64d",&n,&m,&k);
        for(ll i=0;i<n;i++)
        {
            scanf("%I64d",&a[i]);
        }
        memset(dp,0,sizeof(dp));
        for(ll i=0;i<n;i++)
        {

            if(q.size()<k)
            {
                if(a[i]>=m)
                    q.push(i);
                if(q.size()==k)
                    dp[i]=q.front()+1;
                else
                    dp[i]=0;
                continue;
            }
            if(a[i]<m)
            {
                dp[i]=dp[i-1]+1+q.front();
            }
            else
            {
                q.push(i);
                q.pop();
                dp[i]=dp[i-1]+1+q.front();
            }
        }
            printf("%I64d\n",dp[n-1]);

    }
    return 0;
}