RMQ&線段樹複習

     如標題,今天主要複習了RMQ和線段樹,整理了一下模版:

     首先是RMQ:

    

#include<cstdio>  
#include<cstring>  
#include<algorithm>  
using namespace std;  
const int MAXN=50000+100;  
  
int dmax[MAXN][20];  
int dmin[MAXN][20];  
  
void initmax(int n,int d[])//初始化最大值查詢  
{  
    for(int i=1; i<=n; i++)  
        dmax[i][0]=d[i];  
    for(int j=1 ; (1<<j)<=n ; j++)  
        for(int i=1; i+(1<<j)-1 <=n; i++)  
            dmax[i][j]=max(dmax[i][j-1],dmax[i+(1<<(j-1))][j-1]);  
}  
int getmax(int L,int R)//查詢最大值  
{  
    int k=0;  
    while((1<<(k+1))<=R-L+1)k++;  
    return max(dmax[L][k] , dmax[R-(1<<k)+1][k]);  
}  
  
void initmin(int n,int d[])//初始化最小值查詢  
{  
{  
    for(int i=1; i<=n; i++)  
        dmin[i][0]=d[i];  
    for(int j=1; (1<<j)<=n; j++)  
        for(int i=1;i+(1<<j)-1<=n;i++)  
            dmin[i][j]= min( dmin[i][j-1],dmin[i+(1<<(j-1))][j-1] );  
}  
int getmin(int L,int R)//查詢最小值  
{  
    int k=0;  
    while( (1<<(k+1)) <=R-L+1)k++;  
    return min(dmin[L][k],dmin[R-(1<<k)+1][k]);  
}  

線段樹:

   線段樹單點add,區間sum查詢的模板(HDU1166題)如下:
   其實所有的單點更新,區間查詢的題目都是類似的,可以類似解決。

  

#include<cstdio>  
#include<cstring>  
using namespace std;  
const int maxn=50000+5;  
  
//線段樹需要維護的信息  
int sum[maxn*4];  
#define lson i*2, l, m  
#define rson i*2+1, m+1, r  
  
//i節點收集子節點的統計結果  
void PushUp(int i)  
{  
    sum[i]=sum[i*2]+sum[i*2+1];  
}  
  
//遞歸建立線段樹  
void build(int i,int l,int r)  
{  
    if(l==r)  
    {  
        scanf("%d",&sum[i]);  
        return ;  
    }  
  
    int m=(l+r)/2;  
    build(lson);  
    build(rson);  
    PushUp(i);//收集子節點的結果  
}  
  
//在當前區間[l, r]內查詢區間[ql, qr]間的目標值  
//且能執行這個函數的前提是:[l,r]與[ql,qr]的交集非空  
//其實本函數返回的結果也是 它們交集的目標值  
int query(int ql,int qr,int i,int l,int r)  
{  
    //目的區間包含當前區間  
    if(ql<=l && r<=qr) return sum[i];  
  
    int m=(l+r)/2;  
    int res=0;  
    if(ql<=m) res += query(ql,qr,lson);  
    if(m<qr) res += query(ql,qr,rson);  
    return res;  
}  
  
//update這個函數就有點定製的意味了  
//本題是單點更新,所以是在區間[l,r]內使得第id數的值+val  
//如果是區間更新,可以update的參數需要將id變爲ql和qr  
void update(int id,int val,int i,int l,int r)  
{  
    if(l==r)  
    {  
        sum[i] += val;  
        return ;  
    }  
  
    int m=(l+r)/2;  
    if(id<=m) update(id,val,lson);  
    else update(id,val,rson);  
    PushUp(i);//時刻記住維護i節點統計信息正確性  
}  
  
  
int main()  
{  
    int T;  
    scanf("%d",&T);  
    for(int kase=1; kase<=T; kase++)  
    {  
        printf("Case %d:\n",kase);  
  
        int n;//節點總數  
        scanf("%d",&n);  
        build(1,1,n);//建立線段樹  
  
        char str[20];  
        int u,v;  
        while(scanf("%s",str)==1 && str[0]!='E')  
        {  
            scanf("%d%d",&u,&v);  
  
            if(str[0]=='Q') printf("%d\n",query(u,v,1,1,n));  
            else if(str[0]=='A') update(u,v,1,1,n);  
            else update(u,-v,1,1,n);  
        }  
    }  
    return 0;  
}  
題目沒多看,主要是複習了一下原理和模版,明天再複習下DP。
發佈了101 篇原創文章 · 獲贊 23 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章