由于近期参加的两场比赛中均出现了关于线段树以及二叉索引树这样的题,所以就仔细的看了刘汝佳的那本经典的白书(厚版),看懂了原理,就写了最基本的实现代码用来验证的
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define MAX 1000000
int A[MAX];//存储每个结点的值
int C[MAX];//辅助数组
int n;
//定义lowbit操作
int lowbit(int i)
{
return i&-i;
}
//求前x个结点的累积和
int sum(int x)
{
int ret=0;
while(x>0){
ret+=C[x];
x-=lowbit(x);
}
return ret;
}
//如果修改了某个结点值之后需要更新的数组C的值,也用来初始化C数组
void add(int x,int d)
{
while(x<=n){
C[x]+=d;
x+=lowbit(x);
}
}
int main()
{
memset(A,0,sizeof(A));
memset(C,0,sizeof(C));
printf("输入共有多少个数:\n") ;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&A[i]);
add(i,A[i]);//初始化C数组
}
printf("打印出C数组的所有值:\n")
for(int i=1;i<=n;i++)
printf("%d ",C[i]);
printf("\n");
int m;
printf("输入要计算的前m项和:\n");
scanf("%d",&m);
printf("%d\n",sum(m));
return 0;
}