树状数组

#include<iostream>
#include<cstring>
#include<stdio.h>

using namespace std; 

typedef long long ll;

// 空间要开够...... 
// 如果实在装不下, 就得将稀疏数据收紧, 还是不行, 就线段树吧... 
const ll MAXN = 1e6+9;
ll a[MAXN];
ll c[MAXN];
ll ret[MAXN];
ll n;

// int lowbit(int x) {return x&(x^(x–1));}

// 注意 x 不能为 0, 因为 0&0 = 0
int lowbit(int x) {return x&(-x);}

// 前 n 项和 
int Sum(int n)
{
    int s = 0;
    while(n > 0)
    {
        s +=  c[n];
        n -= lowbit(n);
    }
    return s;
 } 

// 修改节点值 
void change(int i, int x)
{    // 居然是这个 MAXN 在作怪..., 算了, 先背下来 
    while(i <= MAXN)
    {    
        c[i] += x;
        i += lowbit(i);
    }
}

// 查询区间 (l-1, r) 的和：区间和 
int find_interval(int l, int r)
{ 
    return Sum(r) - Sum(l-1);
}

void fun(int n)
{
    memset(ret, 0, sizeof(ret));
    memset(c, 0, sizeof(c));
    for(int i=1;i<=n;i++){
        scanf("%d", &a[i]);
        change(a[i], i);
    }
    for(int i=1;i<=n;i++)
        cout<<Sum(i)<<endl;
    return;
}

int main()
{
    while(cin>>n)
    { 
        fun(n);    
    }
    
    return 0;
}

---

2019-02-22

21:07:22