SCUT - 337 - 岩殿居蟹 - 线段树 - 树状数组
这个东西是个阶梯状的。那么可以考虑存两棵树，一棵树是阶梯的，另一棵树的平的，随便一减就是需要的阶梯。
优化之后貌似速度比树状数组还惊人。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
inline int read() {
int x=0;
int f=0;
char c;
do {
c=getchar();
if(c=='-')
f=1;
} while(c<'0'||c>'9');
do {
x=(x<<3)+(x<<1)+c-'0';
c=getchar();
} while(c>='0'&&c<='9');
return f?-x:x;
}
inline void _write(int x) {
if(x>9)
_write(x/10);
putchar(x%10+'0');
}
inline void write(int x) {
if(x<0) {
putchar('-');
x=-x;
}
_write(x);
putchar('\n');
}
void TestCase(int ti);
int main() {
#ifdef Yinku
freopen("Yinku.in","r",stdin);
//freopen("Yinku.out","w",stdout);
#endif // Yinku
int T=1;
for(int ti=1; ti<=T; ti++)
TestCase(ti);
}
/*--- ---*/
const int MAXM=100000;
int a[MAXM+5];
int st[(MAXM<<2)+5];
int st2[(MAXM<<2)+5];
const int mod=1000000007;
inline int add(const int &a,const int &b){
int res=a+b;
return res>=mod?res-mod:res;
}
inline int sub(const int &a,const int &b){
int res=a-b;
return res<0?res+mod:res;
}
inline int mul(const int &a,const int &b){
ll res=1ll*a*b;
return res>=mod?res%mod:res;
}
inline void push_up(int o) {
st[o]=add(st[o<<1],st[o<<1|1]);
st2[o]=add(st2[o<<1],st2[o<<1|1]);
}
void build(int o,int l,int r) {
if(l==r){
st[o]=a[l];
st2[o]=mul(l,a[l]);
}
else {
int m=(l+r)>>1;
build(o<<1,l,m);
build(o<<1|1,m+1,r);
push_up(o);
}
}
void update(int o,int l,int r,int x,int v) {
if(l==r) {
st[o]=v;
st2[o]=mul(x,v);
return;
} else {
int m=(l+r)>>1;
if(x<=m)
update(o<<1,l,m,x,v);
else if(x>=m+1)
update(o<<1|1,m+1,r,x,v);
push_up(o);
}
}
int query1(int o,int l,int r,int a,int b) {
if(a<=l&&r<=b) {
return st[o];
} else {
int m=(l+r)>>1;
int ans=0;
if(a<=m)
ans=query1(o<<1,l,m,a,b);
if(b>=m+1)
ans=add(ans,query1(o<<1|1,m+1,r,a,b));
return ans;
}
}
int query2(int o,int l,int r,int a,int b) {
if(a<=l&&r<=b) {
return st2[o];
} else {
int m=(l+r)>>1;
int ans=0;
if(a<=m)
ans=query2(o<<1,l,m,a,b);
if(b>=m+1)
ans=add(ans,query2(o<<1|1,m+1,r,a,b));
return ans;
}
}
inline void TestCase(int ti) {
int n,m;
while(~scanf("%d",&n)) {
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
build(1,1,n);
scanf("%d",&m);
for(int i=1; i<=m; i++) {
char a[2];
int b,c;
scanf("%s%d%d",a,&b,&c);
if(a[0]=='Q') {
printf("%d\n",sub(query2(1,1,n,b,c),mul(b-1,query1(1,1,n,b,c))));
} else {
update(1,1,n,b,c);
}
}
}
}
这个树状数组就不太好懂了，不过空间是线段树的1/4，速度是其两倍。
单点改值就把差值update上去就可以了。然后记得把原始值也顺手改了。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
inline int read() {
int x=0;
int f=0;
char c;
do {
c=getchar();
if(c=='-')
f=1;
} while(c<'0'||c>'9');
do {
x=(x<<3)+(x<<1)+c-'0';
c=getchar();
} while(c>='0'&&c<='9');
return f?-x:x;
}
inline void _write(int x) {
if(x>9)
_write(x/10);
putchar(x%10+'0');
}
inline void write(int x) {
if(x<0) {
putchar('-');
x=-x;
}
_write(x);
putchar('\n');
}
void TestCase(int ti);
int main() {
#ifdef Yinku
freopen("Yinku.in","r",stdin);
//freopen("Yinku.out","w",stdout);
#endif // Yinku
int T=1;
for(int ti=1; ti<=T; ti++)
TestCase(ti);
}
/*--- ---*/
const int MAXM=100000;
int a[MAXM+5];
int bit[MAXM+5];
int bit2[MAXM+5];
int n;
const int mod=1000000007;
inline int add(const int &a,const int &b) {
int res=a+b;
return res>=mod?res-mod:res;
}
inline int sub(const int &a,const int &b) {
int res=a-b;
return res<0?res+mod:res;
}
inline int mul(const int &a,const int &b) {
ll res=1ll*a*b;
return res>=mod?res%mod:res;
}
inline int sum(int x,int bit[]) {
int res=0;
while(x) {
res=add(res,bit[x]);
x-=x&-x;
}
return res;
}
inline void update(int x,int v,int bit[]) {
while(x<=n) {
bit[x]=add(bit[x],v);
x+=x&-x;
}
}
inline int range_sum(int x,int y,int bit[]) {
return sub(sum(y,bit),sum(x-1,bit));
}
inline void TestCase(int ti) {
int m;
while(~scanf("%d",&n)) {
memset(bit,0,sizeof(bit));
memset(bit2,0,sizeof(bit2));
for(int i=1; i<=n; i++) {
scanf("%d",&a[i]);
update(i,a[i],bit);
update(i,mul(i,a[i]),bit2);
}
scanf("%d",&m);
for(int i=1; i<=m; i++) {
char s[2];
int b,c;
scanf("%s%d%d",s,&b,&c);
if(s[0]=='Q') {
printf("%d\n",sub(range_sum(b,c,bit2),mul(b-1,range_sum(b,c,bit))));
} else {
int delta=sub(c,a[b]);
update(b,delta,bit);
update(b,mul(b,delta),bit2);
a[b]=c;
}
}
}
}
