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=;
 #define mod 1000000007

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

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

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

 int binarySearch(int L,int R,int n,int k)
 {
     int mid;
     int l=L,r=R;
     while(l<=r)
     {
         mid=(l+r)/;
         if(RMQ(mid,R)>=k)
         {
             l=mid+;
         }
         else
         {
             r=mid-;
         }
     }
     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=;i<=n;i++)
                 scanf("%d",&a[i]);
             if(!k)
             {
                 printf("0\n");
                 continue;
             }
             if(k==)
             {
                 printf("%d\n",n);
                 continue;
             }
             init(n);
             ans=;
             for(i=j=;i<=n;i++)
             {
                 j=binarySearch(j,i,n,k);
                 ans+=i-j+;
             }
             printf("%I64d\n",ans);
         }
     }
     return ;
 }

（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=;
 #define mod 1000000007

 int a[];
 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=;i<n;i++)
                 scanf("%d",&a[i]);
             if(!k)
             {
                 printf("0\n");
                 continue;
             }
             if(k==)
             {
                 printf("%d\n",n);
                 continue;
             }
             l=;
             r=;
             multiset<int>s;
             s.insert(a[]);
             ans=n;
             while()
             {
                 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 ;
 }