Fenwick tree can do this job.
class NumArray { vector<int> in; vector<int> ft; int query(int i) { i += 1; i = min(i, int(ft.size() - 1)); int s = 0; for (; i > 0; i -= (i & -i)) s += ft[i]; return s; } void init(int i, int v) { i += 1; for (; i < ft.size(); i += (i & -i)) ft[i] += v; } public: NumArray(vector<int> &nums) { in = nums; int n = nums.size(); ft.assign(n + 1, 0); for(int i = 0; i < n; i ++) init(i, nums[i]); } void update(int i, int v) { int d = v - in[i]; in[i] = v; i += 1; for (; i < ft.size(); i += (i & -i)) ft[i] += d; } int sumRange(int i, int j) { return query(j) - (i ? query(i - 1) : 0); } };
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步