关闭

解题报告:区间求和 I

311人阅读 评论(0) 收藏 举报
http://www.lintcode.com/zh-cn/problem/interval-sum/
/**
 * Definition of Interval:
 * classs Interval {
 *     int start, end;
 *     Interval(int start, int end) {
 *         this->start = start;
 *         this->end = end;
 *     }
 */
 
class Solution { 
public:
    /**
     *@param A, queries: Given an integer array and an query list
     *@return: The result list
     */
    	vector<long long> intervalSum(vector<int> &A, vector<Interval> &queries) {
		// write your code here
		SegmentTreeNode * root = build(0, A.size() - 1, A);

		vector<long long>  d;
		for (int i = 0; i<queries.size(); i++){
			d.push_back(query(root, queries[i].start, queries[i].end));
		}
		return d;
	}
    SegmentTreeNode * build(int start, int end, vector<int> &A) {
		// write your code here
		if (start>end)return NULL;
		int mid = start + (end - start) / 2;
		if (start  == end)return new SegmentTreeNode(start, end, A[start] );
		SegmentTreeNode *left = build(start, mid, A);
		SegmentTreeNode *right = build(mid + 1, end, A);
		SegmentTreeNode *stn = new SegmentTreeNode(start, end, left->max + right->max);
		stn->left = left;
		stn->right = right;
		return stn;
	}
 unsigned int query(SegmentTreeNode *root, int start, int end) {
		// write your code here
		if (root == nullptr)return 0;
		SegmentTreeNode *left = root->left, *right = root->right;
		if (end<root->start)return 0;//end超出了最左
		if (start>root->end)return 0;//start超出了最右
		if (start<root->start)start = left->start;//start超出了最左
		if (end>root->end)end = right->end;//end超出了最右
		if (start == root->start&&end == root->end)return root->max;
		if (left->end >= end){//完全左侧
			return query(left, start, end);
		}
		if (right->start <= start){//完全右侧
			return query(right, start, end);
		}
		return  query(left, start, left->end)+query(right, right->start, end);//左右之间都有
	}
};

0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:55869次
    • 积分:1507
    • 等级:
    • 排名:千里之外
    • 原创:101篇
    • 转载:1篇
    • 译文:0篇
    • 评论:8条
    最新评论