HDU 5289 Assignment （二分+区间最值）

【题目链接】click here~~

【题目大意】：

给出一个数列，问其中存在多少连续子序列，子序列的最大值-最小值<k

【思路】：枚举数列左端点，然后二分枚举右端点，用ST算法求区间最值。（或用单调队列的思路）

代码：

#include <bits/stdc++.h>using namespace std;const int N=1e5+10;typedef long long LL;#define Max(a,b) a>b?a:b#define Min(a,b) a>b?b:a#define mem(a,b) memset(a,b,sizeof(a))int arr[N];int n,k,m,tmp;int dp_max[N][20],dp_min[N][20];void rmq_init(){    for(int i=1; i<=n; ++i) dp_max[i][0]=dp_min[i][0]=arr[i];    double limit=log(n)/log(2.0);  // 换底公式    for(int j=1; j<=(int)limit; ++j){        for(int i=1; i+(1<<j)-1<=n; ++i){            dp_max[i][j]=Max(dp_max[i][j-1],dp_max[i+(1<<(j-1))][j-1]);            dp_min[i][j]=Min(dp_min[i][j-1],dp_min[i+(1<<(j-1))][j-1]);        }    }}int rmq_max(int L,int R){   // 查询[L,R]之间的最大值    int k=floor(log2((double)(R-L+1)));    return Max(dp_max[L][k], dp_max[R - (1<<k) + 1][k]);}int rmq_min(int L, int R){   // 查询[L,R]之间的最小值    int k=floor(log2((double)(R-L+1)));    return Min(dp_min[L][k], dp_min[R - (1<<k) + 1][k]);}int solve(int L,int R){    int k=floor(log2((double)(R-L+1)));    int maxx=Max(dp_max[L][k],dp_max[R - (1<<k) + 1][k]);    int minn=Min(dp_min[L][k],dp_min[R - (1<<k) + 1][k]);    return maxx-minn;}inline LL read(){    int c=0,f=1;    char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}    return c*f;}int main(){    int t;t=read();    while(t--){        n=read();k=read();        for(int i=1; i<=n; ++i){            arr[i]=read();        }        rmq_init();        LL ans=0;        for(int i=1; i<=n; ++i){//枚举左端点，二分右端点,ST求最值            int ll=i,rr=n;            while(ll<=rr){                int mid=(ll+rr)>>1;                if(solve(i,mid)>=k) rr=mid-1;                else ll=mid+1;            }            if(solve(i,rr)<k)ans+=(rr-i+1);            else ans+=(ll-i+1);        }        printf("%I64d\n",ans);    } return 0;}

单调队列：

【思路】：

O(n)复杂度
用两个单调队列维护最大值，最小值，相当于双指针，初始，第一个第二个指针指向第一个数据，第一个指针按顺序不断向队尾添加数据，当最大值最小值的差大于等于k后，意味着新添加的这个不能作用于当前第二个指针的位置，也就能计算出，以第二个指针位置开始的连续子序列的个数，最后统计就可以了。

代码：

#include <bits/stdc++.h>using namespace std;const int N=1e5+10;typedef long long LL;#define Max(a,b) a>b?a:b#define Min(a,b) a>b?b:a#define mem(a,b) memset(a,b,sizeof(a))int arr[N];int i,j,n,k,m,tmp;deque <int >deq_max,deq_min;// maxvalue minvalueinline LL read(){    int c=0,f=1;    char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}    return c*f;}int main(){    int t;t=read();    while(t--){        n=read();k=read();        for(int i=0; i<n; ++i){            arr[i]=read();        }        LL ans=0;        while(!deq_max.empty()) deq_max.pop_back();        while(!deq_min.empty()) deq_min.pop_back();        for(i=0,j=0; i<n; ++i){            while(!deq_max.empty()&&deq_max.back()<arr[i]) deq_max.pop_back();deq_max.push_back(arr[i]);            while(!deq_min.empty()&&deq_min.back()>arr[i]) deq_min.pop_back();deq_min.push_back(arr[i]);            while(!deq_max.empty()&&!deq_min.empty()&&deq_max.front()-deq_min.front()>=k){                ans+=(i-j);                if(deq_max.front()==arr[j]) deq_max.pop_front();                if(deq_min.front()==arr[j]) deq_min.pop_front();                j++;            }        }        while(j<n){            ans+=(i-j);            j++;        }        printf("%I64d\n",ans);    } return 0;}