bestcoder #86 第三题 NanoApe Loves Sequence Ⅱ hdu 5806

NanoApe Loves Sequence Ⅱ

Accepts: 374

Submissions: 946

Time Limit: 4000/2000 MS (Java/Others)

Memory Limit: 262144/131072 K (Java/Others)

问题描述

退役狗 NanoApe 滚回去学文化课啦！

在数学课上，NanoApe 心痒痒又玩起了数列。他在纸上随便写了一个长度为 $n$ 的数列，他又根据心情写下了一个数 $m$。

他想知道这个数列中有多少个区间里的第 $k$ 大的数不小于 $m$，当然首先这个区间必须至少要有 $k$ 个数啦。

输入描述

第一行为一个正整数 $T$，表示数据组数。

每组数据的第一行为三个整数 $n,m,k$。

第二行为 $n$ 个整数 AiA_iA​i​​，表示这个数列。

1≤T≤10, 2≤n≤200000, 1≤k≤n/2, 1≤m,Ai≤1091 \le T \le 10,~2 \le n \le 200000,~1 \le k \le n/2,~1 \le m,A_i \le 10^91≤T≤10, 2≤n≤200000, 1≤k≤n/2, 1≤m,A​i​​≤10​9​​

输出描述

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

输入样例

1
7 4 2
4 2 7 7 6 5 1

输出样例

18

思路：所有区间总数是n的平方，就算对于每个区间只用O(1）的时间也会超时。

然后转化思路：对于数列里的每个元素i （1：n） 如果以它为起点，那么符合要求的区间的可以是从i开始经过 k或k以上个不小于m的数的位置就是终点（显然只要有k个不小于m的数之后的位置都满足）。

那么只要预处理出所有不小于m的数的位置，按序放在一个新的数组里。那么对于每一个i，只要累计它前面不小于m的数的个数，就能O(1)的时间内找到可以为区间终点的最小的值。之后累加每个i得出的值就是答案了！

注意long long

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=200005;
int num[N];
int is[N];
int main()
{
    //freopen("s.txt", "r", stdin);
    int t,n,m,k;
    long long a;
    scanf("%d",&t);
    while(t--)
    {
        int number=0,now=0,pre=1;
        long long sum=0;
        scanf("%d %d %d",&n,&m,&k);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
            if(num[i]>=m)
            {
                number++;
                is[number]=i;
             //   cout<<is[number]<<endl;
            }
          //  cout<<i<<endl;
        }
        is[number+1]=n;
        for(int i=1;i<=n;i++)
        {
            if(num[i]>=m)
            {
                now++;
                if(now+k-1<=number)
                {
                    sum+=1LL*pre*(n-is[now+k-1]+1);
                }
                pre=1;
            }
            else
            {
                pre++;
            }
        }
        cout<<sum<<endl;
    }
    return 0;
}