基于线段树的RMQ

来源:互联网 发布:logistic回归 知乎 编辑:程序博客网 时间:2024/05/07 10:02
const int maxn=1<<17;
int n;
int dat[2&maxn-1];
void Init (int n_)
{
    n=1;
    while(n<n_)n*=2;//为了简单起见,将元素的个数扩大到2的幂;
    for(int i=0; i<2*n-1; i++) dat[i]=INT_MAX; //把所有值都设置为INT_MAX


}
void update(int k,int a) // 把第k个数的值转换为a
{
    k+=n-1;//叶子节点
    dat[k]=a;
    //向上更新
    while(k>0)
    {
        k=(k-1)/2;
        dat[k]=min(dat[k*2+1],dat[k*2+2]);
    }
}
//求【a,b)的最小值
//后面的参数是为了计算方便而传入的
//k是节点的编号,l,r表示这个节点对应的是【l,r)区间
//外部调用时,用query(a,b,0,0,n);
int query(int a,int b,int k,int l,int r)
{
    //如果区间无交集,则返回一个无关的值
    if(r<=a||b<=l) return INT_MAX;
    //如果【l,r)被完全包含则直接返回当前节点的值
    if(a<=l&&r<=b) return dat[k];
    //否则返回两个儿子中较小的那个
    else
    {
        int vl=query(a,b,k*2+1,l,(l+r)/2);
        int vr=query(a,b,k*2+2,(l+r)/2,r);
        return min(vl,vr);
    }
}
0 0
原创粉丝点击