hdu 5289 Assignment(2015多校第一场第2题)RMQ+二分(或者multiset模拟过程)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5289

题意:给你n个数和k,求有多少的区间使得区间内部任意两个数的差值小于k,输出符合要求的区间个数

思路:求出区间的最大最小值,只要他们的差值小于k,那么这个区间就符合要求,但是由于n较大,用暴力一定超时,所以就要用别的方法了;而RMQ是可以求区间的最值的,而且预处理的复杂度只有O(nlogn),而查询只是O(1)处理,这样相对来说节约了时间,再根据右端点来二分枚举左端点(其实不用二分好像更快,估计是数据问题)。

而另一种方法不得不说,学了C++的一定要在认真的去看一下STL的那些用法,这次用multiset来做就大大的节约时间,而且又不用思考太多,只要从左到右模拟区间,就可以了。


(RMQ+二分)代码:

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <iostream>
#include <queue>
#include <algorithm>
#include <vector>
#include <set>
using namespace std;
#define LL __int64
#define INF 0x3f3f3f3f
const int MAXN=100005;
#define mod 1000000007

int a[MAXN];
LL ans;
int dp1[MAXN][30],dp2[MAXN][30];

void init(int n)
{
    for(int i=1;i<=n;i++)
        dp1[i][0]=dp2[i][0]=a[i];
    for(int j=1;(1<<j)<=n;j++)
    {
        for(int i=1;i+(1<<j)-1<=n;i++)
        {
            dp1[i][j]=max(dp1[i][j-1],dp1[i+(1<<(j-1))][j-1]);
            dp2[i][j]=min(dp2[i][j-1],dp2[i+(1<<(j-1))][j-1]);
        }
    }
}

int RMQ(int L,int R)
{
    int k=0;
    while((1<<(k+1))<=R-L+1)k++;
    return max(dp1[L][k],dp1[R-(1<<k)+1][k])-min(dp2[L][k],dp2[R-(1<<k)+1][k]);
}

int binarySearch(int L,int R,int n,int k)
{
    int mid;
    int l=L,r=R;
    while(l<=r)
    {
        mid=(l+r)/2;
        if(RMQ(mid,R)>=k)
        {
            l=mid+1;
        }
        else
        {
            r=mid-1;
        }
    }
    if(RMQ(mid,R)>=k)
        mid++;
    return mid;
}
int main()
{
    int T,i,j,n,k,mi,ma,l,r;
    while(~scanf("%d",&T))
    {
        while(T--)
        {
            scanf("%d%d",&n,&k);
            for(i=1;i<=n;i++)
                scanf("%d",&a[i]);
            if(!k)
            {
                printf("0\n");
                continue;
            }
            if(k==1)
            {
                printf("%d\n",n);
                continue;
            }
            init(n);
            ans=0;
            for(i=j=1;i<=n;i++)
            {
                j=binarySearch(j,i,n,k);
                ans+=i-j+1;
            }
            printf("%I64d\n",ans);
        }
    }
    return 0;
}

(STL)代码:

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <iostream>
#include <queue>
#include <algorithm>
#include <vector>
#include <set>
using namespace std;
#define LL __int64
#define INF 0x3f3f3f3f
const int MAXN=100005;
#define mod 1000000007

int a[100005];
LL ans;
int main()
{
    int T,i,j,n,k,mi,ma,l,r;
    while(~scanf("%d",&T))
    {
        while(T--)
        {
            scanf("%d%d",&n,&k);
            for(i=0;i<n;i++)
                scanf("%d",&a[i]);
            if(!k)
            {
                printf("0\n");
                continue;
            }
            if(k==1)
            {
                printf("%d\n",n);
                continue;
            }
            l=0;
            r=1;
            multiset<int>s;
            s.insert(a[0]);
            ans=n;
            while(1)
            {
                if(s.size())
                {
                    mi=*s.begin();
                    ma=*s.rbegin();
                    if(abs(a[r]-mi)<k&&abs(a[r]-ma)<k)
                    {
                        ans+=s.size();
                        s.insert(a[r]);
                        r++;
                        if(r==n)
                            break;
                    }
                    else
                    {
                        if(s.size())
                            s.erase(s.find(a[l]));
                        l++;
                    }
                }
                else
                {
                    l=r;
                    s.insert(a[r]);
                    r++;
                    if(r==n)
                        break;
                }
            }
            printf("%I64d\n",ans);
        }
    }
    return 0;
}


你可能感兴趣的:(STL,RMQ,二分)