我写的是线段树,维护每组磁铁的 x+⌊x/3⌋ ,区间修改、区间和查询,但是全WA。有没有大佬帮忙看看是哪里错了。
代码
#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[500005];
struct node
{
long long s,tag;
int s0,s1,s2;
}tree[2000005];
void add(int p,int l,int r,long long v)
{
if(v==0)
{
return;
}
if(v>0)
{
tree[p].tag+=v;
tree[p].s+=(v+v/3)*(r-l+1);
if(v%3==1)
{
tree[p].s+=tree[p].s2;
swap(tree[p].s1,tree[p].s2);
swap(tree[p].s0,tree[p].s1);
}
if(v%3==2)
{
tree[p].s+=tree[p].s1+tree[p].s2;
swap(tree[p].s0,tree[p].s1);
swap(tree[p].s1,tree[p].s2);
}
}
else
{
v=-v;
tree[p].tag-=v;
tree[p].s-=(v+v/3)*(r-l+1);
if(v%3==1)
{
tree[p].s-=tree[p].s0;
swap(tree[p].s0,tree[p].s1);
swap(tree[p].s1,tree[p].s2);
}
if(v%3==2)
{
tree[p].s-=tree[p].s0+tree[p].s1;
swap(tree[p].s1,tree[p].s2);
swap(tree[p].s0,tree[p].s1);
}
}
}
void pushup(int p)
{
tree[p].s=tree[p*2].s+tree[p*2+1].s;
tree[p].s0=tree[p*2].s0+tree[p*2+1].s0;
tree[p].s1=tree[p*2].s1+tree[p*2+1].s1;
tree[p].s2=tree[p*2].s2+tree[p*2+1].s2;
}
void pushdown(int p,int l,int r)
{
if(tree[p].tag)
{
int mid=(l+r)>>1;
add(p*2,l,mid,tree[p].tag);
add(p*2+1,mid+1,r,tree[p].tag);
tree[p].tag=0;
}
}
void build(int p,int l,int r)
{
if(l==r)
{
tree[p].s=tree[p].s0=tree[p].s1=tree[p].s2=0;
tree[p].s=a[l]+a[l]/3;
if(a[l]%3==0)
{
tree[p].s0++;
}
if(a[l]%3==1)
{
tree[p].s1++;
}
if(a[l]%3==2)
{
tree[p].s2++;
}
return;
}
int mid=(l+r)>>1;
build(p*2,l,mid);
build(p*2+1,mid+1,r);
pushup(p);
}
void change(int p,int l,int r,int x,int y,int z)
{
if((x<=l)&&(r<=y))
{
add(p,l,r,z);
return;
}
if((y<l)||(r<x))
{
return;
}
int mid=(l+r)>>1;
change(p*2,l,mid,x,y,z);
change(p*2+1,mid+1,r,x,y,z);
pushup(p);
}
long long query(int p,int l,int r,int x,int y)
{
if((x<=l)&&(r<=y))
{
return tree[p].s;
}
if((y<l)||(r<x))
{
return 0;
}
pushdown(p,l,r);
int mid=(l+r)>>1;
return query(p*2,l,mid,x,y)+query(p*2+1,mid+1,r,x,y);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
build(1,1,n);
while(m--)
{
int op;
scanf("%d",&op);
if(op==1)
{
int l,r,v;
scanf("%d%d%d",&l,&r,&v);
change(1,1,n,l,r,v);
}
else
{
int l,r;
scanf("%d%d",&l,&r);
printf("%lld\n",query(1,1,n,l,r));
}
}
}