Question

这是使用分段树的rmq。但我没有得到正确的输出，任何人都可以告诉我我错在哪里 `

#include<iostream>
 #include<cmath>
 #define inf 1e9
using namespace std;
int get_mid(int a,int b){
    return a+(b-a)/2;
}
int min(int a,int b){
    int temp=(a<b)?a:b;
    return temp;
}
int get_min_util(int str[],int beg,int end,int l,int r,int i){
    if(l<=beg && r>=end)
    return str[i];
    if(l>end || r<beg){
        return 0; ////////c1
      }
    int mid=get_mid(beg,end);
    return (get_min_util(str,beg,mid,l,r,2*i+1),get_min_util(str,mid+1,end,l,r,2*i+2));
}
int get_min(int str[],int len,int l,int r){
    if(l<0 || r>len-1 || l>r){
        return 0;
    }
    else
    return get_min_util(str,0,len-1,l,r,0);
}
int build_rmq_arr_util(int str[],int beg,int end,int big_arr[],int i){
    if(beg==end){
        big_arr[i]=str[beg];
        return big_arr[i];
    }
    int mid=get_mid(beg,end);
    big_arr[i]=min(build_rmq_arr_util(str,beg,mid,big_arr,2*i+1),build_rmq_arr_util(str,mid+1,end,big_arr,2*i+2));
    return big_arr[i];
}
int*bulid_rmq_arr(int str[],int len){
    int temp=ceil(log2(len));
    temp=2*pow(2,temp)-1;
    int *big_arr=new int[temp];
    build_rmq_arr_util(str,0,len-1,big_arr,0);
    return big_arr;
}
void pop(int arr[],int size){
    for(int i=0;i<size;i++){
        cout<<arr[i]<<" ";
    }
    cout<<endl;
}
int main(){
    int str[]={1,2,3,4,5,6};
    int len=sizeof(str)/sizeof(int);
    int *rmq_arr;
    rmq_arr=bulid_rmq_arr(str,len);
    //pop(rmq_arr,3*len);
    cout<<get_min(rmq_arr,len,1,2);
}

`
输出此代码：
0
1-2范围内的预期输出为2 函数get_min_util从写入注释//////// c1的条件返回0

Answer 1

问题在于get_min_util（）。当它不在范围内时它应该返回无穷大。你还必须返回最小的左右孩子。

int get_min_util(int str[],int beg,int end,int l,int r,int i){
if(l<=beg && r>=end)
return str[i];
if(l>end || r<beg){
    return inf; ////////c1
  }
int mid=get_mid(beg,end);
return min(get_min_util(str,beg,mid,l,r,2*i+1),get_min_util(str,mid+1,end,l,r,2*i+2));
}

范围最小查询

1 个答案: