线段树练习4

线段树练习4

codevs题号： 4919

时间限制: 1 s

空间限制: 128000 KB

题目描述 Description：

给你N个数，有两种操作

1：给区间[a,b]内的所有数都增加X

2：询问区间[a,b]能被7整除的个数

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是add，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是count，表示统计区间[a,b]能被7整除的个数

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

2 3 4

6

count 1 3

count 1 2

add 1 3 2

count 1 3

add 1 3 3

count 1 3

样例输出 Sample Output

0

0

0

1

数据范围及提示 Data Size & Hint

100%:1< N<= 100000,1< Q<= 100000

题解：

比较水啊这题，我们可以在线段树的结构体数组中加上一个数组g[i]表示这个区间中与7取余为i的数有多少个。每次更新时都更新这个数组的值就好了，具体看代码。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long int a[200010],add[800010];
int n,m,x,y,v,pd,num=0;
struct hh
{
int f,l,r,ls,rs;//ls为左子节点，rs为右子节点
long long int date;
int g[10];//g[i]保存 此区间中和7取余为i的数右多少个
}t[800010];
char pdd[10];
void build(int l,int r)
{

num++;
int i=num;
for(int j=0;j<7;j++)
t[i].g[j]=0;//手动初始化
t[i].l=l;t[i].r=r;
if(l!=r)
{
t[i].ls=num+1;
build(l,(l+r)/2);//建立左子节点

t[i].rs=num+1;
build(((l+r)/2+1),r);//建立右子节点
for(int j=0;j<7;j++)//父节点的g数组的值就是子节点的g数组的和
{
t[i].g[j]=t[t[i].ls].g[j]+t[t[i].rs].g[j];
}

}
else if(l==r)
{
t[i].g[(a[l]%7)]=1;//叶子节点 直接赋值
}

}
void pushdown(int i)//更新子节点的值
{
if(add[i]!=0)
{
int s[10],va=add[i];//利用一个s数组，保存更新后的值
for(int j=0;j<7;j++)
{
s[(j+va)%7]=t[t[i].ls].g[j];//左儿子更新数据
}
for(int j=0;j<7;j++)//再把更新后数据赋给节点的g数组
t[t[i].ls].g[j]=s[j];

for(int j=0;j<7;j++)
{
s[(j+va)%7]=t[t[i].rs].g[j];//右儿子更新数据
}
for(int j=0;j<7;j++)//把更新后数据赋给节点的g数组
t[t[i].rs].g[j]=s[j];

add[t[i].ls]+=va;
add[t[i].rs]+=va;
add[i]=0;
}
}
void update(int i,int l,int r,int value)
{
if(t[i].l==l&&t[i].r==r)//完全覆盖，直接算
{
add[i]+=value;
int s[10];
for(int j=0;j<7;j++)
{
s[(j+value)%7]=t[i].g[j];//利用一个s数组，保存更新后的值
}
for(int j=0;j<7;j++)//再把更新后数据赋给节点的g数组
t[i].g[j]=s[j];
return ;
}
if(t[i].l==t[i].r) return ;
pushdown(i);//把延迟更新的值传给子节点
int z=(t[i].l+t[i].r)/2;//区间更新的套路啦
if(r<=z) update(t[i].ls,l,r,value);
else if(l>z) update(t[i].rs,l,r,value);
else
{
update(t[i].ls,l,z,value);
update(t[i].rs,z+1,r,value);
}
for(int j=0;j<7;j++)//父节点的g数组的值就是子节点的g数组的和
{
t[i].g[j]=t[t[i].ls].g[j]+t[t[i].rs].g[j];
}
}
long long int query(int i,int l,int r)//求和
{
if(t[i].l==l&&t[i].r==r)//完全覆盖，直接返回值
{
return t[i].g[0];
}
pushdown(i);//降延迟更新的值传给儿子节点
int z=(t[i].l+t[i].r)/2;//还是线段树区间更新的套路
if(r<=z) return query(t[i].ls,l,r);
else if(l>z) return query(t[i].rs,l,r);
else return query(t[i].ls,l,z)+query(t[i].rs,z+1,r);
}
int main()
{
memset(add,0,sizeof(add));
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
a[i]%=7; //预处理先 mod7
}
build(1,n);
cin>>m;
for(int i=1;i<=m;i++)
{
scanf("%s",pdd);
{
if(pdd[0]=='a')
{
scanf("%d%d%d",&x,&y,&v);
if((v%=7)!=0)//要加上的值如果和mod7为0的话就没什么用了，不必更新。
update(1,x,y,v%7);
}
else if(pdd[0]=='c')
{
scanf("%d%d",&x,&y);
cout<<query(1,x,y)<<endl;
}
}
}
return 0;
}