您的位置:首页 > 其它

POJ 3468 A Simple Problem with Integers 线段树区间修改

2014-03-16 11:38 423 查看
http://poj.org/problem?id=3468

题目大意:

给你N个数还有Q组操作(1 ≤ N,Q ≤ 100000)

操作分为两种,Q A B 表示输出[A,B]的和 C A B X表示把[A,B]的所有数加上X

思路:

线段树的区间修改。。。。。

昨天晚上改了老半天。

然后关机准备睡觉毕竟今天有实验。。去洗个头。。突然有灵感。。急急忙忙的开电脑改了就对了~哈哈哈

PS:POJ AC 100了~ 因为混迹各个OJ,SO才100

用位运算优化*2 1600+MS,不用2200MS。。。差了挺多

顺便测试了下很久以前想的k<<1和k+k的效率,果然是k<<1的效率高,k+k慢了30多MS

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=100000+10;
typedef long long LL;
LL sum[MAXN<<2],add[MAXN<<2];
int a[MAXN];
void build(LL  k,LL  L,LL  R)
{
if(L==R)  sum[k]=a[L];
else
{
LL m=(L+R)>>1;
build(k<<1,L,m);
build((k<<1)+1,m+1,R);
sum[k]=sum[k<<1]+sum[(k<<1)+1];
}
}

void update(LL  k,LL  L,LL  R,LL  a,LL  b,LL  v)
{
sum[k]+=(min(R,b)-max(a,L)+1)*v;
if(a<=L && R<=b) add[k]+=v;
else
{
LL m=(L+R)>>1;
if(a<=m)	update(k<<1,L,m,a,b,v);
if(m<b)	    update((k<<1)+1,m+1,R,a,b,v);
}
}
LL ans;
void query(LL  k,LL L,LL  R,LL  a,LL  b,LL  addv)
{
if(a<=L && R<=b) ans+=sum[k]+(R-L+1)*addv;
else
{
LL m=(L+R)>>1;
if(a<=m)	query(k<<1,L,m,a,b,addv+add[k]);
if(m<b)	   query((k<<1)+1,m+1,R,a,b,addv+add[k]);

}
}

int main()
{
//	freopen("e:\\input.txt","r",stdin);
int n,q;
while(~scanf("%d%d",&n,&q))
{
memset(add,0,sizeof(add));
memset(sum,0,sizeof(sum));

for(int i=1;i<=n;i++)
scanf("%d",&a[i]);

build(1,1,n);

for(int i=0;i<q;i++)
{
char cmd[5];
int a,b,v;
scanf("%s",cmd);
if(cmd[0]=='Q')
{
ans=0;
scanf("%d%d",&a,&b);
query(1,1,n,a,b,0);
printf("%lld\n",ans);
}
else
{
scanf("%d%d%d",&a,&b,&v);
update(1,1,n,a,b,v);
}
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐