zkw线段树-Build(),单点Updata(),单点Query()。

#include<bits/stdc++.h>
using namespace std;
#define maxn 100007
int A[maxn];//原数组,n为原数组元素个数 ,N为扩充元素个数 
int Tree[maxn << 2];//区间和 
int Add[maxn << 2];//懒惰标记 
int N;
void Build(int n) {
    //计算N的值 
    N = 1; while (N < n + 2) N <<= 1;
    //更新叶节点 
    for (int i = 1; i <= n; ++i) Tree[N + i] = A[i];//原数组下标+N=存储下标
    //更新非叶节点 
    for (int i = N - 1; i > 0; --i) {
        //更新所有非叶节点的统计信息 
        Tree[i] = Tree[i << 1] + Tree[i << 1 | 1];
        //清空所有非叶节点的Add标记 
        Add[i] = 0;
    }
}
void Update(int L, int C) {
    for (int s = N + L; s; s >>= 1) {
        Tree[s] += C;
    }
}
int Query(int L, int R) {
    int ANS = 0;
    for (int s = N + L - 1, t = N + R + 1; s^t ^ 1; s >>= 1, t >>= 1) {
        if (~s & 1) ANS += Tree[s ^ 1];
        if (t & 1) ANS += Tree[t ^ 1];
    }
    return ANS;
}
int main()
{
    int n;
    cin >> n;
    for (int i = 1; i <= n; ++i)A[i] = 1;
    Build(n);
    cout << Query(1, n) << endl;
    Update(1, 1);//A[1]=A[1]+1
    Update(2, 1);//A[2]=A[2]+1
    cout << Query(1, n) << endl;
}

猜你喜欢

转载自blog.csdn.net/qq_31741481/article/details/84451988