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;
}