HDU 5289 2015 Multi-University Training Contest 1 1002 RMQ+二分

　　　　　　　　　　　　　　　　　　Assignment

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1758    Accepted Submission(s): 849

Problem Description

Tom owns a company and he is the boss. There are n staffs which are numbered from 1 to n in this company, and every staff has a ability. Now, Tom is going to assign a special task to some staffs who were in the same group. In a group, the difference of the ability of any two staff is less than k, and their numbers are continuous. Tom want to know the number of groups like this.

 

Input

In the first line a number T indicates the number of test cases. Then for each case the first line contain 2 numbers n, k (1<=n<=100000, 0<k<=10^9),indicate the company has n persons, k means the maximum difference between abilities of staff in a group is less than k. The second line contains n integers:a[1],a[2],…,a[n](0<=a[i]<=10^9),indicate the i-th staff’s ability.

 

Output

For each test，output the number of groups.

 

Sample Input

2

4 2

3 1 2 4

10 5

0 3 4 5 2 1 6 7 8 9

 

Sample Output

5 28

Hint

First Sample, the satisfied groups include:[1,1]、[2,2]、[3,3]、[4,4] 、[2,3]

 

Author

FZUACM

 

Source

2015 Multi-University Training Contest 1

 

题意：

给出一个序列，求这个序列里面有多少组，满足组内的最大值-最小值<K

刚开始的想法

设dp[i]表示以i为左端点，最远可以到达的位置，使得区间[i,i],[i,i+1],,,[i,dp[i]]都满足条件

则以i为左端点有dp[i]-i+1个区间满足条件，再对所有的i，累加dp[i]-i+1，即为答案

问题是：递推不出dp[i]

后来看了题解，可以用rmq+二分求出i最远可以到达的位置，而不是递推。

先用st算法求出任意区间的最大值和最小值的差。

枚举左端点i，二分右端点j，则以i为左端点有j-i+1个区间满足。

累加，即为答案。

注意：最后答案要用long long

#include<cstdio>
#include<cstring>

#define MAX(x,y) x>y?x:y;
#define MIN(x,y) x<y?x:y;
#define LL long long

using namespace std;

const int MAXN=100000+10;
int a[MAXN];
int min[MAXN][20];
int max[MAXN][20];

void init(int N)
{
    for(int i=1;i<=N;i++)
    {
        min[i][0]=a[i];
        max[i][0]=a[i];
    }
    for(int j=1;(1<<j)<=N;j++)
    {
        for(int i=1;(i+(1<<j)-1)<=N;i++)
        {
            min[i][j]=MIN(min[i][j-1],min[i+(1<<(j-1))][j-1]);
            max[i][j]=MAX(max[i][j-1],max[i+(1<<(j-1))][j-1]);
        }
    }
}

int query(int l,int r)
{
    int k=0;
    while((1<<(k+1))<=r-l+1)
        k++;
    int x=MAX(max[l][k],max[r-(1<<k)+1][k]);
    int y=MIN(min[l][k],min[r-(1<<k)+1][k]);
    return x-y;
}

LL solve(int N,int K)
{
    LL res=0;
    for(int i=1;i<=N;i++)
    {
        int l=i,r=N;
        while(r-l>1)
        {
            int p=(l+r)>>1;
            if(query(i,p)<K)
                l=p;
            else
                r=p;
        }
        if(query(i,r)<K)
            res+=(LL)(r-i+1);
        else
            res+=(LL)(l-i+1);
    }
    return res;
}

int main()
{
    int test;
    scanf("%d",&test);
    while(test--)
    {
        int N,K;
        scanf("%d%d",&N,&K);
        for(int i=1;i<=N;i++)
        {
            scanf("%d",&a[i]);
        }
        init(N);

        printf("%I64d
",solve(N,K));
    }
    return 0;
}