题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5289
题 意:给T组数据,然后每组一个n和k,表示n个数,k表示最大允许的能力差,接下来n个数表示n个人的能力,求能力差在k之内的区间有几个。
思 路:预处理出区间[i,j]中的最大值最小值(用RMQ算法),然后枚举左端点l,二分出符合条件的右端点r,以l为左端点,j为右端点(l<=j<=r)的符合条件的集合为(r-l+1)。
代码如下:
#include <iostream>
using namespace std;
#include <string.h>
#include <stdio.h>
#include <cmath>
#include <algorithm>
#define maxn 200005
typedef __int64 LL;
int a[maxn];
int mx[maxn][30],mi[maxn][30];
int n, k;
void RMQ()//RMQ算法的处理与存储
{
for( int i = 1; i <= n; i ++ )
mx[i][0] = mi[i][0] = a[i];
for( int j = 1; (1<<j) <= n; j ++ )
{
for( int i = 1; i + (1<<j) - 1 <= n; i ++ )
{
int p = 1<<(j-1);
mx[i][j] = max( mx[i][j-1], mx[i+p][j-1] );
mi[i][j] = min( mi[i][j-1], mi[i+p][j-1] );
}
}
}
int q( int i, int j )//RMQ算法的查询
{
int k = log2( j-i+1.0 );//在区间的长度为j - i + 1的区间内查询最值
int maxx = max( mx[i][k], mx[j-(1<<k)+1][k] );
int minx = min( mi[i][k], mi[j-(1<<k)+1][k] );
return maxx - minx;
}
int main()
{
int T;
scanf ( "%d", &T );
while( T-- )
{
scanf ( "%d %d", &n, &k );
for( int i = 1; i <= n; i ++ )
scanf ( "%d", &a[i] );
RMQ();
LL ans = 0;
for( int i = 1; i <= n; i ++ )
{
int l = i, r = n;
while( l + 1 < r )
{
int m = ( l+r ) / 2;
if( q(i,m) < k ) l = m;//条件成立更新左值
else r = m;
}
//cout<<i<<l<<" "<<""<<""<<""<<endl;
if (q(i, r) < k)
ans += (r * 1LL - i + 1);
else ans += (l * 1LL - i + 1);
}
printf( "%I64d\n", ans );
}
return 0;
}