将区间和s改为s[7] 表示该区间除以7余0–6的数的个数,push_dwon的时候对于每个s[i]进行更新即可
代码如下
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<vector>
#define MAXN 100010
#include<algorithm>
using namespace std;
struct Node{int l,r,ls,rs,s[7],d;};
vector<Node> N;
int n,Q,a[MAXN];
void build_tree(int v){
if(N[v].l==N[v].r){N[v].s[a[N[v].l]%7]++;return;}
int mid=(N[v].l+N[v].r)>>1;
N.push_back((Node){N[v].l,mid,0,0});
N[v].ls=N.size()-1;
N.push_back((Node){mid+1,N[v].r,0,0});
N[v].rs=N.size()-1;
build_tree(N[v].ls),build_tree(N[v].rs);
for(int i=0;i<7;i++) N[v].s[i]=N[N[v].ls].s[i]+N[N[v].rs].s[i];
}
void push_down(int v){
int b[14];
for(int i=0;i<7;i++) b[i]=b[i+7]=N[v].s[i];
N[v].d%=7;
for(int i=0;i<7;i++) N[v].s[i]=b[i+7-N[v].d];
if(N[v].l<N[v].r) N[N[v].ls].d+=N[v].d,N[N[v].rs].d+=N[v].d;
N[v].d=0;
}
void change(int v,int l,int r,int val){
push_down(v);
if(N[v].l>r||N[v].r<l) return;
else if(N[v].l>=l&&N[v].r<=r){
N[v].d+=val;
push_down(v);
}
else{
change(N[v].ls,l,r,val);
change(N[v].rs,l,r,val);
for(int i=0;i<7;i++) N[v].s[i]=N[N[v].ls].s[i]+N[N[v].rs].s[i];
}
}
int Query(int v,int l,int r){
if(N[v].l>r||N[v].r<l) return 0;
push_down(v);
if(N[v].l>=l&&N[v].r<=r) return N[v].s[0];
return Query(N[v].ls,l,r)+Query(N[v].rs,l,r);
}
int main(){
scanf("%d",&n);
N.push_back((Node){1,n,0,0});
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
build_tree(0);
scanf("%d",&Q);
while(Q--){
char Type[7]; scanf("%s",Type);
if(Type[0]=='a'){
int a,b,x; scanf("%d%d%d",&a,&b,&x);
change(0,a,b,x);
}
else{
int a,b; scanf("%d%d",&a,&b);
printf("%d\n",Query(0,a,b));
}
}
return 0;
}