LINK:http://acm.hdu.edu.cn/showproblem.php?pid=1166
比较Naive的线段树题目,最基本的单点刷新,区间查询,注意使用Scanf,不然就会T掉 =。=
不知道自己的算法还是实现什么的哪里有一点问题,最终时间还是跑到了760MS,请指教。
贴代码:
#include
#include
using namespace std;
struct node {
int left,right;
int max;
int sum;
};
int Max(int i,int j){
return i>j?i:j;
}
node tree[200010];
void build(int id,int l,int r){ //建树
tree[id].left = l;
tree[id].right = r;
if (l==r) { // 这是一个叶子节点
scanf("%d",&tree[id].max);
tree[id].sum = tree[id].max;
}
else { // 这是非终结节点
int mid = (l+r)/2;
build (id<<1,l,mid);
build (id<<1|1,mid+1,r);
tree[id].sum = tree[id<<1].sum+tree[id<<1|1].sum;
// tree[id].max = Max (tree[id<<1].max , tree[id<<1|1].max);
}
}
void update(int id,int pos,int val,int tag){ // 在pos位置修改元素val
if (tree[id].left == tree[id].right){
tree[id].sum += val*tag;
// tree[id].max += val*tag;
}
else {
int mid = (tree[id].left + tree[id].right)/2;
if (pos<=mid)
update(id<<1,pos,val,tag);
else update(id<<1|1,pos,val,tag);
tree[id].sum = tree[id<<1].sum + tree[id<<1|1].sum;
// tree[id].max = Max(tree[id<<1].max , tree[id<<1|1].max);
}
}
int query(int id,int l,int r){ // 查询[l,r]中间的总和或者最大值
if (tree[id].left == l && tree[id].right == r) {
return tree[id].sum;
}
else {
int mid = (tree[id].left + tree[id].right)/2;
if (r<=mid)
return query(id<<1,l,r);
else if (l>mid)
return query(id<<1|1,l,r);
else
return query(id<<1,l,mid)+query(id<<1|1,mid+1,r);
}
}
int main(){
int n;
cin>>n;
for (int i=1;i<=n;++i) {
cout<<"Case "<":"<<endl;
int l;
scanf("%d",&l);
build(1,1,l);
string instru;int b,e;
while (1){
cin>>instru;
if (instru[0]=='E') break;
scanf("%d%d",&b,&e);
if (instru[0]=='Q'){
cout<<query(1,b,e)<<endl;
}
if (instru[0]=='A'){
update(1,b,e,1);
}
if (instru[0]=='S'){
update(1,b,e,-1);
}
}
}
}