poj2352 Stars （第一道树状数组

树状数组详解

通过这道题，我对树状数组的理解：

树状数组通过一种类似与二叉树的结构，使得任意连续段的求和操作的时间复杂度为O(log(N)).但是付出的代价是，对某一个元素做修改的时候的时间复杂度不再是O(1),而是O(log(N)).

一般数组求某一连续段的和的时间复杂度为O(N),而树状数组为O(log(N)). 因此，树状数组适合于，需要大量经常计算某连续段的和的情景。

#include<stdio.h> #include<ctype.h> #define lowbit(t) (t&(t^(t-1))) #define RESULT_MAX 15000 /* */ #define MAX_NUM 32005 /* */ //please declare parameters here. int in[MAX_NUM]; int result[RESULT_MAX]; int N; //please declare functions here. //求前n项和 int sum(int end) { int sum = 0; while(end > 0) { sum += in[end]; end -= lowbit(end); } return sum; } //增加某个元素的大小 void plus(int pos, int num) { while(pos<=MAX_NUM) { in[pos] += num; pos += lowbit(pos); } } int main() { if(freopen("input.txt","r",stdin)==NULL) perror("Can not open the input file!"); //input your . scanf("%d\n",&N); int i,x; for(i=0;i<N;i++) { x=0; char ch; while(isdigit((ch=getchar()))) { x=x*10+(ch-'0'); } while(isdigit(getchar())); // scanf("%d %d",&x,&y); // printf("%d %d\n",x,y); x++; result[sum(x)]++; plus(x,1); } for(i=0;i<N;i++) { printf("%d\n",result[i]); } return 0; }