http://acm.hdu.edu.cn/showproblem.php?pid=4027
又是纯粹的线段树的基本操作。
(1)数组data[]是纯是浪费,不用开。数组sum[]在 build() 函数下应保证初始化成功(尤其是指先赋值为0,在读入数据)。
(2)if(b>c) {int x=b;b=c;c=x;} 这关乎AC与否的问题。
(3)稍微注意__int64 。
(4)本题不需要成段更新(也没办法成段更新),更新时肯定会比较耗时。这里有个优化,当所考虑的区间里德只全为1(sum==r-l+1),就不需要更新了。
if(sum[rt]==r-l+1) return ;
具体代码:
View Code
#include<stdio.h> #include<math.h> #define lson l, m, rt<<1 #define rson m+1, r, rt<<1|1 const int maxn=101000; __int64 sum[maxn<<2]; int n, m; void pushup(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void build(int l, int r, int rt) { sum[rt]=0; if(l==r) { scanf("%I64d", &sum[rt]); return ; } int m=(l+r)>>1; build(lson); build(rson); pushup(rt); } __int64 query(int L, int R, int l, int r, int rt) { if(L<=l&&r<=R) return sum[rt]; int m=(l+r)>>1; __int64 ret=0; if(L<=m) ret+=query(L, R, lson); if(R>m) ret+=query(L, R, rson); return ret; } void update(int L, int R, int l, int r, int rt) { if(sum[rt]==r-l+1) return ; if(l==r) { sum[rt]=sqrt((double)sum[rt]); return ; } int m=(l+r)>>1; if(L<=m) update(L, R, lson); if(R>m) update(L, R, rson); pushup(rt); } int main() { int i, j, k; int cas=0; while(scanf("%d", &n)!=EOF) { printf("Case #%d:\n", ++cas); build(1, n, 1); scanf("%d", &m); for(i=1;i<=m;i++) { int a, b, c; scanf("%d%d%d", &a, &b, &c); if(b>c) {int x=b;b=c;c=x;} if(a) { printf("%I64d\n", query(b, c, 1, n, 1)); } else { update(b, c, 1, n, 1); } } printf("\n"); } return 0; }
