先呈上原题链接[HDU-4578]
题意:
对于一个区间有4个操作:
1.将a~b都加上c
2.将a~b都乘上c
3.将a~b都变成c
4.查询a~b的每个数的p次方的和。(p=1,2,3)
思路:
虽说是一道很裸的线段树的题,但是却非常考验代码能力。
查询p的取值只有三个,所以维护三棵线段树就行了。
难点在多种操作,每个操作的更新次序是有讲究的,在下推标记的时候一般是先推置数操作,然后推乘法,最后推加法。
坑点:
一定要注意有没有爆范围。。。小心WA到怀疑人生。
good luck and have fun!!!
附上代码:
#include<bits/stdc++.h>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
#define test(flag,value) cout<<flag<<":"<<(value)<<endl
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> pii;
const int INF=0x3f3f3f3f;
const int MAXN=1e5+10;
const int MOD=10007;
const double PI=acos(-1);
int n,m;
int sum[MAXN<<2][4],lazy[MAXN<<2][4];
LL tmp1,tmp2,tmp3;
void init()
{
for(int i=0;i<=n*4;i++)
{
lazy[i][1]=0;
lazy[i][2]=1;
lazy[i][3]=0;
sum[i][1]=sum[i][2]=sum[i][3]=0;
}
}
void push_up(int rt)
{
for(int i=1;i<=3;i++)
sum[rt][i]=sum[rt<<1][i]+sum[rt<<1|1][i];
}
void push_down(int rt,int l,int r)
{
int m=(l+r)>>1;
if(lazy[rt][3]!=0)
{
tmp1=lazy[rt][3]%MOD;
sum[rt<<1][1]=(tmp1*(m-l+1))%MOD;
sum[rt<<1|1][1]=(tmp1*(r-m))%MOD;
tmp2=(tmp1*lazy[rt][3])%MOD;
sum[rt<<1][2]=(tmp2*(m-l+1))%MOD;
sum[rt<<1|1][2]=(tmp2*(r-m))%MOD;
tmp3=(tmp2*lazy[rt][3])%MOD;
sum[rt<<1]