本文主要是介绍LeetCode 题202:线段树的查询,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目描述:
对于一个有n个数的整数数组,在对应的线段树中, 根节点所代表的区间为0-n-1, 每个节点有一个额外的属性max,值为该节点所代表的数组区间start到end内的最大值。
为SegmentTree设计一个 query 的方法,接受3个参数root, start和end,线段树root所代表的数组中子区间[start, end]内的最大值。
例如:
对于数组 [1, 4, 2, 3], 对应的线段树为:
query(root, 1, 1), return 4
query(root, 1, 2), return 4
query(root, 2, 3), return 3
query(root, 0, 2), return 4
数据结构:
/*** Definition of SegmentTreeNode:* class SegmentTreeNode * {* public:* int start, end, max;* SegmentTreeNode *left, *right;* SegmentTreeNode(int start, int end, int max) * {* this->start = start;* this->end = end;* this->max = max;* this->left = this->right = NULL;* }* }**/
代码:
class Solution
{
public:int query(SegmentTreeNode *root, int start, int end) {if(root == NULL){return NULL;}if(root->start >= start && root->end <= end){return root->max;}int mid = (root->start + root->end) / 2;if(mid < start){return query(root->right, start, end);}else if(mid + 1 > end){return query(root->left, start, end);}else{return max(query(root->right, start, end), query(root->left, start, end));}}
};
这篇关于LeetCode 题202:线段树的查询的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!