nyoj123 士兵杀敌(四)线段树(lazy标记)+树状数组 区间修改 单点查询

区间修改时用lazy标记区间平均增加数,同时更新区间和(r-l+1)*ave,查找某点的值的过程中如果某点有lazy值则向下扩散,并把该点lazy值标0。

 
#include <stdio.h>
const int maxn=1000001*4;
struct node{
	int l,r,lazy,sum;
}n[maxn];
void build(int k,int l,int r){
	n[k].l=l; n[k].r=r;
	if(l==r){
		n[k].lazy=0;
		n[k].sum=0;
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
}
void add(int k,int l,int r,int sum){
	if(n[k].l==l&&n[k].r==r){
		n[k].lazy+=sum;
		n[k].sum+=(r-l+1)*sum;
		return;
	}
	int mid=(n[k].l+n[k].r)>>1;
	if(r<=mid) add(k<<1,l,r,sum);
	else if(l>mid) add(k<<1|1,l,r,sum);
	else{
		add(k<<1,l,mid,sum);
		add(k<<1|1,mid+1,r,sum);
	}
}
int query(int k,int a){
	if(n[k].l==n[k].r){
		return n[k].sum;
	}
	if(n[k].lazy){
		n[k<<1].lazy+=n[k].lazy;
		n[k<<1|1].lazy+=n[k].lazy;
		n[k<<1].sum+=(n[k<<1].r-n[k<<1].l+1)*n[k].lazy;
		n[k<<1|1].sum+=(n[k<<1|1].r-n[k<<1|1].l+1)*n[k].lazy;
		n[k].lazy=0;
	}
	int mid=(n[k].l+n[k].r)>>1;
	if(a<=mid) query(k<<1,a);
	else query(k<<1|1,a);
}
int main(){
	int t,m,a,b,c;
	char s[7];
	scanf("%d %d",&t,&m);
	build(1,1,m);
	while(t--){
	scanf("%s",s);
		if(s[0]=='A'){
			scanf("%d %d %d",&a,&b,&c);
			add(1,a,b,c);
		}
		else{
			scanf("%d",&a);
			printf("%d\n",query(1,a));
		}
	}
}        

 

区间修改,单点查询,query()查询的为单点值。

区间查找,单点修改,query()查询的为前缀和。 

#include <stdio.h>
int l[1000005],m;
int lowbit(int x){
	return x&(-x);
}
void add(int x,int y){
	while(x<=m){
		l[x]+=y;
		x+=lowbit(x);
	}
}
int query(int x){
	int sum=0;
	while(x>0){
		sum+=l[x];
		x-=lowbit(x);
	}
	return sum;
}
int main(){
	int t,a,b,c;
	char s[7];
	scanf("%d %d",&t,&m);
	while(t--){
		scanf("%s",s);
		if(s[0]=='A'){
			scanf("%d %d %d",&a,&b,&c);
			add(a,c);add(b+1,-c);
		}
		else{
			scanf("%d",&a);
			printf("%d\n",query(a));
		}
	}
} 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值