树状数组区间求和

最新推荐文章于 2024-10-27 12:59:29 发布

Colinnian

最新推荐文章于 2024-10-27 12:59:29 发布

阅读量278

点赞数 5

文章标签：算法数据结构

本文链接：https://blog.csdn.net/Colinnian/article/details/137695638

版权

using ll = long long;
template<class T>
struct Fenwick {
	int n;
	std::vector<T>a;
	Fenwick(int n_) {
		init(n_);
	}
	void init(int n_) {
		n = n_;
		a.assign(n, T{});
	}
	void add(int x, const T& v) {
		for (int i = x + 1; i <= n; i += i & -i) {
			a[i - 1] = a[i - 1] + v;
		}
	}
	T sum(int x) {
		T ans{ };
		for (int i = x; i > 0; i -= i & -i) {
			ans = ans + a[i - 1];
		}
		return ans;
	}
	T rangeSum(int l, int r) {
		return sum(r) - sum(l);
	}
};
int main()
{
    std::cin.sync_with_stdio(false);
    int n, m;
    std::cin >> n >> m;
    ll a, b = 0;
    Fenwick<ll> f1(n + 1),f2(n+1);
    for (ll i = 1; i <= n; i++)
    {
        std::cin >> a;
        b = a - b;
        f1.add(i, b);
        f2.add(i, (i - 1) * b);
        b = a;
    }
    for (ll i = 1; i <= m; i++)
    {
        int t, x, y, z;
        std::cin >> t;
        if (t == 1)
        {
            std::cin >> x >> y >> z;
            f1.add(x, z);
            f1.add(y + 1, -z);
            f2.add(x, z * (x - 1));
            f2.add(y + 1, -z * y);//此处为核心，联系上方的公式，想一想为什么这么修改。
        }
        else
        {
            std::cin >> x >> y;
            std::cout << (y * f1.sum(y+1) - (x - 1) * f1.sum(x)) - (f2.sum(y+1) - f2.sum(x)) << '\n';
        }
    }
    return 0;
}