HDU 5289 Assignment (二分+RMQ）

最新推荐文章于 2018-08-26 19:04:18 发布

原创最新推荐文章于 2018-08-26 19:04:18 发布 · 526 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#algorithm

二分同时被 2 个专栏收录

3 篇文章

订阅专栏

RMQ

2 篇文章

订阅专栏

本文介绍了一道编号为 HDU 5289 的 ACM 竞赛题目的解题思路及代码实现。该题要求求解在给定的能力值范围内，有多少个连续的子序列满足特定条件。文章详细介绍了使用 RMQ 算法进行预处理，并通过二分查找确定符合条件的子序列数量的方法。

题目链接：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;
}