HDU4578 线段树(区间更新 + 多种操作)

时间:2023-03-08 16:10:39
HDU4578 线段树(区间更新 + 多种操作)

  题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4578  , 线段树的区间更新 + 多种操作,好题。

  虽然是比较裸的线段树,但是比较麻烦,并且有很多细节需要考虑,最后我7.3s很惊险地过了,求大神告知优化方法。


  这道题坑在有三种询问:set , add , mul。所以lazy标记要有三个,如果三个标记同时出现的处理方法——当更新set操作时,就把add标记和mul标记全部取消;当更新mul操作时,如果当前节点add标记存在,就把add标记改为:add * mul。这样的话就可以在PushDown()操作中先执行set,然后mul,最后add。

  麻烦在有三种询问:和 , 平方和 , 立方和。对于set和mul操作来说,这三种询问都比较好弄。

  对于add操作,和的话就比较好弄,按照正常方法就可以;

  平方和这样来推:(a + c)2 = a2 + c2 + 2ac  , 即sum2[rt] = sum2[rt] + (r - l + 1) * c * c + 2 * sum1[rt] * c;

  立方和这样推:(a + c)3 = a3 + c3 + 3a(a2 + ac) , 即sum3[rt] = sum3[rt] + (r - l + 1) * c * c * c + 3 * c * (sum2[rt] + sum1[rt] * c);

  几个注意点:add标记取消的时候是置0,mul标记取消的时候是置1;在PushDown()中也也要注意取消标记,如set操作中取消add和mul,mul操作中更新add; 在add操作中要注意sum3 , sum2 , sum1的先后顺序,一定是先sum3 , 然后sum2 , 最后sum1; int容易爆,还是用LL要保险一点; 最后就是运算较多,不要漏掉东西。

#include <iostream>
#include <cstdio>
#include <vector>
#include <cmath>
#include <string>
#include <string.h>
#include <algorithm>
using namespace std;
#define LL __int64
#define eps 1e-8
#define INF INT_MAX
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
const int MOD = ;
const int maxn = + ;
const int N = ;
LL add[maxn << ] , set[maxn << ] , mul[maxn << ];
LL sum1[maxn << ] , sum2[maxn << ] , sum3[maxn << ];
void PushUp(int rt)
{
sum1[rt] = (sum1[rt << ] + sum1[rt << | ]) % MOD;
sum2[rt] = (sum2[rt << ] + sum2[rt << | ]) % MOD;
sum3[rt] = (sum3[rt << ] + sum3[rt << | ]) % MOD;
}
void build(int l , int r , int rt)
{
add[rt] = set[rt] = ;
mul[rt] = ;
if(l == r) {
sum1[rt] = sum2[rt] = sum3[rt] = ;
return;
}
int m = (l + r) >> ;
build(lson);
build(rson);
PushUp(rt);
}
void PushDown(int rt , int len)
{
if(set[rt]) {
set[rt << ] = set[rt << | ] = set[rt];
add[rt << ] = add[rt << | ] = ; //注意这个也要下放
mul[rt << ] = mul[rt << | ] = ;
LL tmp = ((set[rt] * set[rt]) % MOD) * set[rt] % MOD;
sum1[rt << ] = ((len - (len >> )) % MOD) * (set[rt] % MOD) % MOD;
sum1[rt << | ] = ((len >> ) % MOD) * (set[rt] % MOD) % MOD;
sum2[rt << ] = ((len - (len >> )) % MOD) * ((set[rt] * set[rt]) % MOD) % MOD;
sum2[rt << | ] = ((len >> ) % MOD) * ((set[rt] * set[rt]) % MOD) % MOD;
sum3[rt << ] = ((len - (len >> )) % MOD) * tmp % MOD;
sum3[rt << | ] = ((len >> ) % MOD) * tmp % MOD;
set[rt] = ;
}
if(mul[rt] != ) { //这个就是mul[rt] != 1 , 当时我这里没注意所以TLE了
mul[rt << ] = (mul[rt << ] * mul[rt]) % MOD;
mul[rt << | ] = (mul[rt << | ] * mul[rt]) % MOD;
if(add[rt << ]) //注意这个也要下放
add[rt << ] = (add[rt << ] * mul[rt]) % MOD;
if(add[rt << | ])
add[rt << | ] = (add[rt << | ] * mul[rt]) % MOD;
LL tmp = (((mul[rt] * mul[rt]) % MOD * mul[rt]) % MOD);
sum1[rt << ] = (sum1[rt << ] * mul[rt]) % MOD;
sum1[rt << | ] = (sum1[rt << | ] * mul[rt]) % MOD;
sum2[rt << ] = (sum2[rt << ] % MOD) * ((mul[rt] * mul[rt]) % MOD) % MOD;
sum2[rt << | ] = (sum2[rt << | ] % MOD) * ((mul[rt] * mul[rt]) % MOD) % MOD;
sum3[rt << ] = (sum3[rt << ] % MOD) * tmp % MOD;
sum3[rt << | ] = (sum3[rt << | ] % MOD) * tmp % MOD;
mul[rt] = ;
}
if(add[rt]) {
add[rt << ] += add[rt]; //add是+= , mul是*=
add[rt << | ] += add[rt];
LL tmp = (add[rt] * add[rt] % MOD) * add[rt] % MOD; //注意sum3 , sum2 , sum1的先后顺序
sum3[rt << ] = (sum3[rt << ] + (tmp * (len - (len >> )) % MOD) + * add[rt] * ((sum2[rt << ] + sum1[rt << ] * add[rt]) % MOD)) % MOD;
sum3[rt << | ] = (sum3[rt << | ] + (tmp * (len >> ) % MOD) + * add[rt] * ((sum2[rt << | ] + sum1[rt << | ] * add[rt]) % MOD)) % MOD;
sum2[rt << ] = (sum2[rt << ] + ((add[rt] * add[rt] % MOD) * (len - (len >> )) % MOD) + ( * sum1[rt << ] * add[rt] % MOD)) % MOD;
sum2[rt << | ] = (sum2[rt << | ] + (((add[rt] * add[rt] % MOD) * (len >> )) % MOD) + ( * sum1[rt << | ] * add[rt] % MOD)) % MOD;
sum1[rt << ] = (sum1[rt << ] + (len - (len >> )) * add[rt]) % MOD;
sum1[rt << | ] = (sum1[rt << | ] + (len >> ) * add[rt]) % MOD;
add[rt] = ;
}
}
void update(int L , int R , int c , int ch , int l , int r , int rt)
{
if(L <= l && R >= r) {
if(ch == ) {
set[rt] = c;
add[rt] = ;
mul[rt] = ;
sum1[rt] = ((r - l + ) * c) % MOD;
sum2[rt] = ((r - l + ) * ((c * c) % MOD)) % MOD;
sum3[rt] = ((r - l + ) * (((c * c) % MOD) * c % MOD)) % MOD;
} else if(ch == ) {
mul[rt] = (mul[rt] * c) % MOD;
if(add[rt])
add[rt] = (add[rt] * c) % MOD;
sum1[rt] = (sum1[rt] * c) % MOD;
sum2[rt] = (sum2[rt] * (c * c % MOD)) % MOD;
sum3[rt] = (sum3[rt] * ((c * c % MOD) * c % MOD)) % MOD;
} else if(ch == ) {
add[rt] += c;
LL tmp = (((c * c) % MOD * c) % MOD * (r - l + )) % MOD; //(r - l + 1) * c^3
sum3[rt] = (sum3[rt] + tmp + * c * ((sum2[rt] + sum1[rt] * c) % MOD)) % MOD;
sum2[rt] = (sum2[rt] + (c * c % MOD * (r - l + ) % MOD) + * sum1[rt] * c) % MOD;
sum1[rt] = (sum1[rt] + (r - l + ) * c) % MOD;
}
return;
}
PushDown(rt , r - l + );
int m = (l + r) >> ;
if(L > m)
update(L , R , c , ch , rson);
else if(R <= m)
update(L , R , c , ch , lson);
else {
update(L , R , c , ch , lson);
update(L , R , c , ch , rson);
}
PushUp(rt);
}
LL query(int L , int R , int p , int l , int r , int rt)
{
if(L <= l && R >= r) {
if(p == )
return sum1[rt] % MOD;
else if(p == )
return sum2[rt] % MOD;
else
return sum3[rt] % MOD;
}
PushDown(rt , r - l + );
int m = (l + r) >> ;
if(L > m)
return query(L , R , p , rson);
else if(R <= m)
return query(L , R , p , lson);
else
return (query(L , R , p , lson) + query(L , R , p , rson)) % MOD;
}
int main()
{
int n , m;
int a , b , c , ch;
while(~scanf("%d %d" , &n , &m))
{
if(n == && m == )
break;
build( , n , );
while(m--) {
scanf("%d %d %d %d" , &ch , &a , &b , &c);
if(ch != ) {
update(a , b , c , ch , , n , );
} else {
printf("%I64d\n" , query(a , b , c , , n , ));
}
}
}
return ;
}