标签:bre sub == add printf clu pre 速度 修改
Input第一行一个整数T,表示有T组数据。
每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
接下来每行有一条命令,命令有4种形式:
(1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30)
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30);
(3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数;
(4)End 表示结束,这条命令在每组数据最后出现;
每组数据最多有40000条命令
Output对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。
Sample Input
1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End
Sample Output
Case 1: 6 33 59
早上听DHL大佬说此题用树状数组也能做,于是花了个把小时研究了一下。比线段树短了将近一倍,速度也快了不少。不得不说树状数组大法好!
树状数组到不是很难理解,但网上的说法有些太简洁,找到了一篇不错的博客,记录一下:http://www.cnblogs.com/wuwangchuxin0924/p/5921130.html
代码:
#include <iostream> #include <stdio.h> #include <memory.h> using namespace std; int n,a[50005]; char sh[15]; int lowbit(int x) //精髓之所在,靠lowbit找到二进制后面有几个0 { return x&(-x); } void update(int i,int val) { while(i<=n) { a[i]+=val; i+=lowbit(i);//逐层向上找当前节点的父节点,修改这些节点 } } int sum(int i) { int sum=0; while(i>0) { sum+=a[i]; i-=lowbit(i);//逐层向下找前一棵树的根节点,相加求和 } return sum; } int main() { int i,val,t,x,y,zz=1; scanf("%d",&t); while(t--) { memset(a,0,sizeof(a)); scanf("%d", &n); for(i = 1; i <= n; i++) { scanf("%d", &val); update(i, val); } printf("Case %d:\n", zz++); while(scanf("%s", sh)) { if(sh[0] == ‘E‘) break; scanf("%d %d", &x, &y); if(sh[0] == ‘A‘) update(x, y); else if(sh[0] == ‘S‘) update(x, -y); else printf("%d\n", sum(y)-sum(x-1)); //区间相减求和 } } return 0; }
标签:bre sub == add printf clu pre 速度 修改
原文地址:http://www.cnblogs.com/stxy-ferryman/p/7182470.html