hdu 4578 Transformation 线段树多种操作裸题

时间:2021-05-03 15:46:14

自己写了一个带结构体的WA了7.8次 但是测了几组小数据都对。。感觉问题应该出在模运算那里。写完这波题解去对拍一下。

以后线段树绝不写struct!一般的struct都带上l,r 但是一条线段的长度确定的话,每一个节点的l,r都可以确定的,没必要用struct存上,如果不带上l,r那不是更没必要用结构体hhh

这道题有点坑的地方就是顺序问题

pushdown就是一个最大的难点

change表示改数字 add表示加上数字 multiply表示乘以一个数字

正确的方法应该是 先change 再multiply 再add

先看multiply和add的关系

现在p的数字是x 先add a 然后multiply b 就会变成 b*(x+a)假如我们采用保留括号的方法 那么先multiply后add会变得异常麻烦

因为先multiply后add是b*x+a 再搞出一个括号来会有误差 所以我们选择不保留括号 如果先add后multiply的话 add的数字就乘上multiply的数字

这样就不会有误差 所以在pushdown操作里面 我们就先multiply后add 因为多项式是bx+a嘛 肯定是先乘后加

再看change操作

假如现在p有change标记以及multiply || add操作(我们已经搞清楚multiply和add操作了,就可以把他们看成一个操作ma就好了)

如果先change后ma 我们对lp、rp是不是就得先change后ma 相当于把 x1 x2都改成a 再乘4 相当于x1改成a乘4 x2改成a乘4

如果先ma后change 我们是不是ma的标记全被清空了 这样先change的话也没问题 因为对lp rp根本不会进行ma操作,因为标记都清空了

具体对于每个操作

change就是区间长度乘以对应的x的次数咯

multiply就是原来的和乘以对应的x的次数咯

add的话就要用到一点点初中还是高中知识

一次的话就没得说 原来的和加上区间长度乘以x

二次三次见下述公式

$\begin{aligned}\left( x_{1}+a\right) ^{2}+\left( x_{2}+a\right) ^{2}+\ldots +\left( x_{n}+a\right)^{2} = \left( x^{2}_{i}+x^{2}_{2}+\ldots +x^{2}_{n}\right) +2a\left( x_{1}+x_{2}+\ldots +x_{n}\right) +na^{2}\end{aligned}$

$\begin{aligned}\left( x_{1}+a\right) ^{3}+\left( x_{2}+a\right) ^{3}+\ldots +\left( x_{n}+a\right)^{3} = \left( x^{3}_{i}+x^{3}_{2}+\ldots +x^{3}_{n}\right) +3a^{2}\left( x_{1}+x_{2}+\ldots +x_{n}\right) + 3a\left( x^{2}_{1}+x^{2}_{2}+\ldots +x^{2}_{n}\right) +na^{2}\end{aligned}$

提取一下公因数

$\begin{aligned}\left( x_{1}+a\right) ^{3}+\left( x_{2}+a\right) ^{3}+\ldots +\left( x_{n}+a\right)^{3} = \left( x^{3}_{i}+x^{3}_{2}+\ldots +x^{3}_{n}\right) +3a\left( x^{2}_{1}+x^{2}_{2}+\ldots +x^{2}_{n}+ a\left(x_{1} + x_{2} + \ldots + x_{n}\right)\right) +na^{2}\end{aligned}$

因为求三次需要二次的和 求二次需要一次的和 那么就先算3次再算2次最后算1次

模运算部分因为太多项了。所以犯晕。保险的办法就是加括号后加一个%MOD 这样就比较保险

add和change清除后是0,multiply清楚后是1

代码如下

#include <cstdio>
#include <cstring>
#include <algorithm>
#define lp p<<1
#define rp p<<1|1
#define ll long long
#define MOD 10007
using namespace std;
const int maxn = 1e5 + ; ll add[maxn<<], multiply[maxn<<], change[maxn<<];
ll sum[maxn<<][];
inline void pushup(int p) {
for (int i = ; i <= ; i++) {
sum[p][i] = (sum[lp][i] + sum[rp][i]) % MOD;
}
}
void pushdown(int p, int llen, int rlen) {
if (change[p]) {
change[lp] = change[rp] = change[p];
multiply[lp] = multiply[rp] = ;
add[lp] = add[rp] = ;
sum[lp][] = (((((llen * change[p] % MOD) % MOD) * change[p] % MOD) % MOD) * change[p] % MOD) % MOD;
sum[rp][] = (((((rlen * change[p] % MOD) % MOD) * change[p] % MOD) % MOD) * change[p] % MOD) % MOD;
sum[lp][] = (((llen * change[p] % MOD) % MOD) * change[p] % MOD) % MOD;
sum[rp][] = (((rlen * change[p] % MOD) % MOD) * change[p] % MOD) % MOD;
sum[lp][] = (llen * change[p] % MOD) % MOD;
sum[rp][] = (rlen * change[p] % MOD) % MOD;
change[p] = ;
}
if (multiply[p] != ) {
if (add[lp]) add[lp] = add[lp] * multiply[p] % MOD;
if (add[rp]) add[rp] = add[rp] * multiply[p] % MOD;
multiply[lp] = multiply[lp] * multiply[p] % MOD;
multiply[rp] = multiply[rp] * multiply[p] % MOD;
sum[lp][] = ((((((sum[lp][] % MOD) * multiply[p] % MOD) % MOD) * multiply[p] % MOD) % MOD) * multiply[p] % MOD) % MOD;
sum[rp][] = ((((((sum[rp][] % MOD) * multiply[p] % MOD) % MOD) * multiply[p] % MOD) % MOD) * multiply[p] % MOD) % MOD;
sum[lp][] = ((((sum[lp][] % MOD) * multiply[p] % MOD) % MOD) * multiply[p] % MOD) % MOD;
sum[rp][] = ((((sum[rp][] % MOD) * multiply[p] % MOD) % MOD) * multiply[p] % MOD) % MOD;
sum[lp][] = ((sum[lp][] % MOD) * multiply[p] % MOD) % MOD;
sum[rp][] = ((sum[rp][] % MOD) * multiply[p] % MOD) % MOD;
multiply[p] = ;
}
if (add[p]) {
add[lp] = (add[lp] + add[p]) % MOD;
add[rp] = (add[rp] + add[p]) % MOD;
int temp = ((add[p] * add[p] % MOD) * add[p]) % MOD;
sum[lp][] = ((sum[lp][] + (temp * llen) % MOD) % MOD + (( * add[p] % MOD) * (sum[lp][] + sum[lp][] * add[p]) % MOD)) % MOD;
sum[rp][] = ((sum[rp][] + (temp * rlen) % MOD) % MOD + (( * add[p] % MOD) * (sum[rp][] + sum[rp][] * add[p]) % MOD)) % MOD;
sum[lp][] = (sum[lp][] + ((llen * add[p] % MOD) * add[p] % MOD) + (( * add[p] * sum[lp][]) % MOD)) % MOD;
sum[rp][] = (sum[rp][] + ((rlen * add[p] % MOD) * add[p] % MOD) + (( * add[p] * sum[rp][]) % MOD)) % MOD;
sum[lp][] = (sum[lp][] + ((llen * add[p]) % MOD)) % MOD;
sum[rp][] = (sum[rp][] + ((rlen * add[p]) % MOD)) % MOD;
add[p] = ;
}
}
void build(int p, int l, int r) {
multiply[p] = ;
add[p] = change[p] = ;
if (l == r) {
sum[p][] = sum[p][] = sum[p][] = ;
return;
}
int mid = l + r >> ;
build(lp, l, mid);
build(rp, mid + , r);
pushup(p);
}
void update(int p, int l, int r, int x, int y, int z, int type) {
if (x <= l && y >= r) {
if (type == ) {
add[p] += z; add[p] %= MOD;
sum[p][] = (sum[p][] + ((((((r - l + ) * z) % MOD) * z) % MOD) * z) % MOD + (( * z) % MOD) * (sum[p][] + (sum[p][] * z) % MOD)) % MOD;
sum[p][] = (sum[p][] + ((((r - l + ) * z) % MOD) * z) % MOD + ( * z * sum[p][] % MOD)) % MOD;
sum[p][] = (sum[p][] + ((r - l + ) * z) % MOD) % MOD;
} else if (type == ) {
multiply[p] = multiply[p] * z % MOD;
if (add[p]) add[p] = add[p] * z % MOD;
sum[p][] = (((((sum[p][] * z) % MOD) * z) % MOD) * z) % MOD;
sum[p][] = (((sum[p][] * z) % MOD) * z) % MOD;
sum[p][] = (sum[p][] * z) % MOD;
} else if (type == ) {
change[p] = z;
add[p] = ;
multiply[p] = ;
sum[p][] = ((((((r - l + ) * z) % MOD) * z) % MOD) * z) % MOD;
sum[p][] = ((((r - l + ) * z) % MOD) * z) % MOD;
sum[p][] = ((r - l + ) * z) % MOD;
}
return;
}
int mid = l + r >> ;
pushdown(p, mid - l + , r - mid);
if (x <= mid) update(lp, l, mid, x, y, z, type);
if (y > mid) update(rp, mid + , r, x, y, z, type);
pushup(p);
}
ll query(int p, int l, int r, int x, int y, int z) {
if (x <= l && y >= r) return sum[p][z] % MOD;
int mid = l + r >> ;
pushdown(p, mid - l + , r - mid);
ll res = ;
if (x <= mid) res = (res + query(lp, l, mid, x, y, z)) % MOD;
if (y > mid) res = (res + query(rp, mid + , r, x, y, z)) % MOD;
return res % MOD;
}
int main() {
int n, m;
while (~scanf("%d%d", &n, &m) && n || m) {
build(, , n);
int t, x, y, c;
while (m--) {
scanf("%d%d%d%d", &t, &x, &y, &c);
if (t == ) {
printf("%lld\n", query(, , n, x, y, c));
} else {
update(, , n, x, y, c, t);
}
}
}
return ;
}