codvs4919线段树练习4

题目链接

将区间和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;
}