307. Range Sum Query - Mutable

最新推荐文章于 2021-07-06 17:43:54 发布

qq_34229391

最新推荐文章于 2021-07-06 17:43:54 发布

阅读量106

点赞数

文章标签： leetcode vector

本文链接：https://blog.csdn.net/qq_34229391/article/details/81079845

版权

该博客主要讨论LeetCode中的307题，涉及如何在给定整数数组nums中，实现对元素范围i到j（包括i和j）的和进行查询。提供了update函数来修改数组元素，并强调了数组更新后的range查询策略，即通过维护一个辅助vector num，使得num[i]表示nums前i项的和，从而能快速计算range并处理边界条件。

摘要由CSDN通过智能技术生成

Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.

The update(i, val) function modifies nums by updating the element at index i to val.

Example:

Given nums = [1, 3, 5]

sumRange(0, 2) -> 9
update(1, 2)
sumRange(0, 2) -> 8

Note:

The array is only modifiable by the update function.
You may assume the number of calls to update and sumRange function is distributed evenly.

vector<int> num，num[i]意义是nums的第0项到第i项的和，计算range(i,j)时只需要计算num[j]-num[i-1]，要考虑边界情况，并且在更新数组中某个位置处的值后，应该更新对应的num

Code:

#include <iostream>
#include <cstdlib>
#include <string>
#include <vector>
using namespace std;
class NumArray
{
private:
	vector<int> num;
public:
	NumArray(vector<int> nums)
	{
		num.resize(nums.size(), 0);
		if (nums.size() > 0)num[0] = nums[0];
		for (int i = 1; i < nums.size(); i++)num[i] = num[i - 1] + nums[i];
	}
	void update(int i, int val)
	{
		int cha = (i == 0 ? 0 : num[i - 1]) - num[i] + val;
		for (int a = i; a < num.size(); a++)num[a] += cha;
	}
	int sumRange(int i, int j)
	{
		if (i == 0)return num[j];
		return num[j] - num[i - 1];
	}
};
int main()
{
	vector<int> nums{ 9, - 8 };
	NumArray* obj = new NumArray(nums);
	int param_1 = obj->sumRange(1, 1);
	obj->update(0, 3);
	int param_2 = obj->sumRange(0, 1);
	cout << param_1 << " " << param_2 << endl;
	system("pause");
	return 0;
}