1. 程式人生 > >快速查詢區間最值——RMQ演算法(線段樹實現程式碼)

快速查詢區間最值——RMQ演算法(線段樹實現程式碼)

要求找出區間內的最大最小值的差。

#include<stdio.h>
#include<string.h>
#include<math.h>
#define lson l,m,p<<1
#define rson m+1,r,p<<1|1
#define Max(a,b) (a<b?b:a)
#define Min(a,b) (a<b?a:b)
#define INF 999999999

/*線段樹
int N,M;
int MinP[50000*4+10],MaxP[50000*4+10];
int minT,maxT;

void Update(int
val,int K,int l,int r,int p){ int m=(l+r)>>1; if(l==r){ MaxP[p]=MinP[p]=val; return; } if(K<=m) Update(val,K,lson); else Update(val,K,rson); MaxP[p]=Max(MaxP[p<<1],MaxP[p<<1|1]); MinP[p]=Min(MinP[p<<1],MinP[p<<1
|1]); } void Query(int L,int R,int l,int r,int p){ int m=(l+r)>>1; if(L<=l&&r<=R){ minT=Min(minT,MinP[p]); maxT=Max(maxT,MaxP[p]); return; } if(L<=m) Query(L,R,lson); if(R>=m+1) Query(L,R,rson); } int main(){ int
i,val,a,b; scanf("%d %d",&N,&M); for(i=1;i<=N;i++){ scanf("%d",&val); Update(val,i,1,N,1); } for(i=1;i<=M;i++){ scanf("%d %d",&a,&b); minT=INF; maxT=0; Query(a,b,1,N,1); printf("%d\n",maxT-minT); } return 0; } ST(動態規劃)*/ int N,M; int A[50001]; int FMin[50001][20],FMax[50001][20]; void Init(){ int i,j; for(i=1;i<=N;i++) FMin[i][0]=FMax[i][0]=A[i]; for(i=1;(1<<i)<=N;i++){ //按區間長度遞增順序遞推 for(j=1;j+(1<<i)-1<=N;j++){ //區間起點 FMin[j][i]=Min(FMin[j][i-1],FMin[j+(1<<(i-1))][i-1]); FMax[j][i]=Max(FMax[j][i-1],FMax[j+(1<<(i-1))][i-1]); } } } int Query(int l,int r){ int k=(int)(log(r-l+1)/log(2)); return Max(FMax[l][k],FMax[r-(1<<k)+1][k])-Min(FMin[l][k],FMin[r-(1<<k)+1][k]); } int main(){ int i,a,b; scanf("%d %d",&N,&M); for(i=1;i<=N;i++) scanf("%d",&A[i]); Init(); for(i=1;i<=M;i++){ scanf("%d %d",&a,&b); printf("%d\n",Query(a,b)); } return 0; }