社区讨论
样例已过 0pts,求调
P3373【模板】线段树 2参与者 2已保存回复 2
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 2 条
- 当前快照
- 1 份
- 快照标识符
- @m2ir5kpt
- 此快照首次捕获于
- 2024/10/21 16:26 去年
- 此快照最后确认于
- 2025/11/04 16:37 4 个月前
bro刚学线段树想尝试自己打一遍代码
CPP#include <cstdio>
#define int long long
using namespace std;
const int N=1e5+10;
int mod;
int tree[N<<2];
int tag1[N<<2],tag2[N<<2];
int a[N];
int n,q;
int LS(int x) {return x<<1;}
int RS(int x) {return x<<1|1;}
void addtag1(int l,int r,int i,int k) //写的时候忘了
{
if(!tag1[i]) tag1[i]=k;
else tag1[i]*=k;
tree[i]=(k%mod)*(tree[i]%mod);//相当于总和也乘K
}
void addtag2(int l,int r,int i,int k)
{
tag2[i]+=k;
tree[i]+=k*(r-l+1)%mod;
tree[i]%=mod;
}
void push_up(int i)
{
tree[i]=(tree[LS(i)]+tree[RS(i)])%mod;
}
void push_down(int l,int r,int i)
{
int mid=l+r>>1;
if(tag1[i])
{
addtag1(l,mid,LS(i),tag1[i]); //第三个参数序号开始错了
addtag1(mid+1,r,RS(i),tag1[i]);
tag1[i]=0; //别忘清零
}
if(tag2[i])
{
addtag2(l,mid,LS(i),tag2[i]);
addtag2(mid+1,r,RS(i),tag2[i]);
tag2[i]=0;
}
}
void build(int l,int r,int i)
{
if(l==r)
{
tree[i]=a[l];
return;
}
int mid=l+r>>1;
build(l,mid,LS(i));
build(mid+1,r,RS(i));
push_up(i);
}
int query(int L,int R,int pl,int pr,int p)
{
if(L<=pl && pr<=R) return tree[p]%mod;
push_down(pl,pr,p);
int mid=pl+pr>>1;
int ans=0;
if(L<=mid) ans+=query(L,R,pl,mid,LS(p))%mod;
if(R>mid) ans+=query(L,R,mid+1,pr,RS(p))%mod;
return ans%mod;
}
void update1(int L,int R,int pl,int pr,int p,int k)
{
if(L<=pl && pr<=R)
{
addtag1(pl,pr,p,k);
return;
}
int mid=pl+pr>>1;
push_down(pl,pr,p);//别忘了先传tag
if(L<=mid) update1(L,R,pl,mid,LS(p),k);
if(R>mid) update1(L,R,mid+1,pr,RS(p),k);
push_up(p);//别忘了把值传上来!!
}
void update2(int L,int R,int pl,int pr,int p,int k)
{
if(L<=pl && pr<=R)
{
addtag2(pl,pr,p,k);
return;
}
int mid=pl+pr>>1;
push_down(pl,pr,p);//别忘了先传tag
if(L<=mid) update2(L,R,pl,mid,LS(p),k);
if(R>mid) update2(L,R,mid+1,pr,RS(p),k);
push_up(p);//别忘了把值传上来
}
signed main()
{
scanf("%lld%lld%lld",&n,&q,&mod);
for(int i=1;i<=n;i++) scanf("%lld",a+i);
build(1,n,1);
int op,x,y,k;
while(q--)
{
scanf("%lld",&op);
switch(op)
{
case 1:
{
scanf("%lld%lld%lld",&x,&y,&k);
update1(x,y,1,n,1,k);
}
break;
case 2:
{
scanf("%lld%lld%lld",&x,&y,&k);
update2(x,y,1,n,1,k);
}
break;
case 3:
{
scanf("%lld%lld",&x,&y);
printf("%lld\n",query(x,y,1,n,1));
}
}
}
return 0;
}
回复
共 2 条回复,欢迎继续交流。
正在加载回复...