P3924 康娜的线段树（线段树+期望）

题目

$题目传送门$
小林是个程序媛，不可避免地康娜对这种人类的“魔法”产生了浓厚的兴趣，于是小林开始教她OI。

今天康娜学习了一种叫做线段树的神奇魔法，这种魔法可以维护一段区间的信息，是非常厉害的东西。康娜试着写了一棵维护区间和的线段树。由于她不会打标记，因此所有的区间加操作她都是暴力修改的。具体的代码如下：

struct Segment_Tree{
#define lson (o<<1)
#define rson (o<<1|1)
    int sumv[N<<2],minv[N<<2];
    inline void pushup(int o){sumv[o]=sumv[lson]+sumv[rson];}
    inline void build(int o,int l,int r){
        if(l==r){sumv[o]=a[l];return;}
        int mid=(l+r)>>1;
        build(lson,l,mid);build(rson,mid+1,r);
        pushup(o);
    }
    inline void change(int o,int l,int r,int q,int v){
        if(l==r){sumv[o]+=v;return;}
        int mid=(l+r)>>1;
        if(q<=mid)change(lson,l,mid,q,v);
        else change(rson,mid+1,r,q,v);
        pushup(o);
    }
}T; 

在修改时，她会这么写：

for(int i=l;i<=r;i++)T.change(1,1,n,i,addv);

显然，这棵线段树每个节点有一个值，为该节点管辖区间的区间和。

康娜是个爱思考的孩子，于是她突然想到了一个问题：

如果每次在线段树区间加操作做完后，从根节点开始等概率的选择一个子节点进入，直到进入叶子结点为止，将一路经过的节点权值累加，最后能得到的期望值是多少？

康娜每次会给你一个值 qwq ，保证你求出的概率乘上 qwq 是一个整数。

这个问题太简单了，以至于聪明的康娜一下子就秒了。

现在她想问问你，您会不会做这个题呢？

输入输出格式
输入格式：
第一行整数 n,m,qwq表示线段树维护的原序列的长度，询问次数，分母。

第二行 n 个数，表示原序列。

接下来 m 行，每行三个数 l,r,x表示对区间[l,r] 加上 x
输出格式：
共 m 行，表示期望的权值和乘上qwq结果。
输入输出样例
输入样例#1：

8 2 1
1 2 3 4 5 6 7 8
1 3 4
1 8 2

输出样例#1：

90
120

题解

• 朴素的想法：设最终结果为$ans$，每一层的点权和为$va{l}_i$，$maxD$为最大层数，那么$$ans=\sum _i^{maxD}va{l}_i\ast \frac{1}{2^{i-1}}$$
• 然后画个图我们可以发现，从根节点到叶子节点的路径有且只有一条，那么我们可以方便的维护每个叶子节点的期望值，设从根节点到叶子节点的权值和为$su{m}_i$，每一个节点的深度为$d_i$，每个节点的期望为$an{s}_i$，那么$$an{s}_i=su{m}_i\ast \frac{1}{2^{d_i-1}}$$
• 然后我们可以维护一个深度的前缀和我们就可以$O(1)$回答询问了。
• 对于上面的式子$$an{s}_i=su{m}_i\ast \frac{1}{2^{d_i-1}}$$
• 每次记录时都除以$2^{d_i-1}$非常麻烦，所以我们可以化简一下$$an{s}_i=su{m}_i\ast \frac{1}{2^{d_i-1}}=su{m}_i\ast \frac{2^{maxD}}{2^{d_i-1+maxD}}=su{m}_i\ast \frac{2^{maxD-d_i}}{2^{maxD-1}}$$
• 我们在记录时可以直接记录$su{m}_i\ast 2^{maxD-d_i}$，最后直接除以$2^{maxD-1}$即可
• 对于每次修改来说因为每个点的权值都会改变，设增加的数为$x$，那么增加的期望值为$$x\ast \sum _i^{dep}\frac{1}{2^{i-1}}=x\ast \frac{2^{dep}-1}{2^{dep-1}}=x\ast \frac{(2^{dep}-1)\ast 2^{maxD-dep}}{2^{maxD-1}}$$
• 可以发现上边两个式子的分母都是$2^{maxD-1}$，我们在统计答案时直接算就好了
• 其实瞎搞一下就可以了

code

#include <bits/stdc++.h> 
using namespace std; 
typedef long long LL; 
const int maxn = 1e6 + 100; 

template <typename T> 
inline void read(T &s) {
	s = 0; 
	T w = 1, ch = getchar(); 
	while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); }
	while (isdigit(ch)) { s = (s<<1) + (s<<3) + (ch^48); ch = getchar(); }
	 s *= w;  
}

LL n, m, qwq, maxD, Div, ans;  
LL a[maxn], dep[maxn], sum[maxn]; 
struct node {
	LL l, r; 
	LL sum, add; 
} t[maxn * 4]; 

inline void build(LL p, LL l, LL r, LL d) {
	t[p].l = l, t[p].r = r; 
	if (l == r) {
		t[p].sum = a[l]; 
		dep[l] = d; 
		maxD = max(maxD, d); 
		return ; 
	}
	LL mid = (l + r)>>1; 
	build(p<<1, l, mid, d + 1); 
	build(p<<1|1, mid + 1, r, d + 1); 
	t[p].sum = t[p<<1].sum + t[p<<1|1].sum; 
}

inline LL query(LL p, LL l, LL r, LL d, LL sum) {
	if (l == r) {
		return (1<<d) * (sum + t[p].sum); 
	}
	else {
		int mid = (l + r)>>1; 
		return query(p<<1, l, mid, d - 1, sum + t[p].sum) + query(p<<1|1, mid + 1, r, d - 1, sum + t[p].sum); 
	}
}

inline LL gcd(LL x, LL y) {
	return y ? gcd(y, x % y) : x; 
}

int main() {
	read(n), read(m), read(qwq); 
	for (int i = 1; i <= n; ++i) read(a[i]); 
	build(1, 1, n, 1); 
	ans = query(1, 1, n, maxD - 1, 0); 
	Div = 1<<(maxD - 1); 
	LL g = gcd(qwq, Div); 
	Div /= g; qwq /= g; 
	for (int i = 1; i <= n; ++i) {
		sum[i] = sum[i - 1] + (((1<<dep[i]) - 1) << (maxD - dep[i])); 
	}
	for (int i = 1; i <= m; ++i) {
		LL l, r, x; 
		read(l), read(r), read(x); 
		ans += (sum[r] - sum[l-1]) * x; 
		printf("%lld\n", ans / Div * qwq); 
	}
	return 0; 
}