标签:node index arr start 求和 线段 tree int 区间
#include <bits/stdc++.h> using namespace std; //以下是线段树的模板,让区间查询和修改的时间复杂度到O(lgn); class XianDuanTree{ private: vector<int> arr; vector<int> tree; public: XianDuanTree(vector<int> &arr){ this->arr = arr; } //建树,根据vector<int> arr;构建区间查询树vector<int>tree //start,end限定区间范围,node_index是这个结点在tree的索引; void build_tree(vector<int> &arr,vector<int> &tree,int start,int end,int node_index ){ //只有一个值了,即叶子结点 if(start == end){ tree[node_index] = arr[start]; return; } int left_index = 2*node_index+1; int right_index = 2*node_index+2; int mid = start+(end-start)/2; //递归; build_tree(arr,tree,start,mid,left_index); build_tree(arr,tree,mid+1,end,right_index); //后序位置 tree[node_index] = tree[left_index]+tree[right_index]; } vector<int> ues_build(){ int n = arr.size(); tree.resize(2*n,0); build_tree(arr,tree,0,n-1,0); return tree; } //查询区间和 int query(vector<int> & arr,vector<int> & tree,int start,int end,int left,int right,int node_index){ if(left > end || right < start) return 0; else if(left<=start && end <= right) return tree[node_index]; else{ int mid = start+(end-start)/2; int left_node = 2*node_index+1; int right_node = 2*node_index+2; int left_sum = query(arr,tree,start,mid,left,right,left_node); int right_sum = query(arr,tree,mid+1,end,left,right,right_node); int sum = left_sum+right_sum; return sum; } } int ues_query(int left,int right){ return query(arr,tree,0,arr.size()-1,left,right,0); } //更新单点更新,arr数组 //upadte_index为需要修改的arr中的数值下标,把值改成val void update(vector<int> & arr,vector<int> & tree,int start,int end,int node_index,int update_index,int val){ if(start == end) { arr[start] = val; tree[node_index] = arr[start]; } else{ int left_node = 2*node_index+1; int right_node = 2*(node_index+1); int mid = start+(end-start)/2; if(update_index <= mid){ update(arr,tree,start,mid,left_node,update_index,val); }else{ update(arr,tree,mid+1,end,right_node,update_index,val); } tree[node_index] = tree[left_node]+tree[right_node]; } } void use_update(int index,int val){ update(arr,tree,0,arr.size()-1,0,index,val); } }; int main() { vector<int> arr={93,90,50,50,1}; XianDuanTree * xian = new XianDuanTree(arr); vector<int> tree = xian->ues_build(); xian->use_update(4,2); for (int a:tree) { cout << a << " " <<endl; } int sum = xian->ues_query(2,4); return 0; }
标签:node,index,arr,start,求和,线段,tree,int,区间 来源: https://www.cnblogs.com/cotoyo25/p/16347895.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。