树状数组

中心思想是lowbit函数，

int lowbit(int x){
  return x&(-x);
}

树状数组的修改与查询

int tree[maxn]
void update(int x,int d){/// 修改元素ax ax=ax+d
  while (x < maxn){///maxn维护的总长度
    tree[x] += d; ///当前用到这个数的数+d
    x +=lowbit(x); ///移动到下一个用到这个数的数
  }
}
///查询：
int query(int x){///查询 前缀和 从a1+a2+....+ax
  int ans=0;
  while (x > 0){///找到左边界
    ans += tre[x];///加上这一层的数
    x -= lowbit(x);///移动到下一个层
  }
}

函数使用方法：

利用update()可以实现我们初始化，也可以修改元素，原理都一样都是修改元素。

查询利用query()直接查询前缀和

修改和查询的复杂度都是$O(logn)$.

区间修改+单点查询

单点修改+区间查询直接可以用上面的。下面我们看区间修改利用到差分数组。

区间修改：修一个区间使得一个区间同时+d

其实就是利用一个查分的思想来做,上面的操作是单点修改+区间查询实质是最前缀和数组进行的操作，我们将前缀和进行差分对差分后的数组进行操作这样就想到与对区间修改。

放上一个题：当然这个题可以直接用差分来写，代码和理解起来都更容易些。

///区间修改+单点查询
#include <bits/stdc++.h>
using namespace std;

const int maxn=1e5+7;

int tree[maxn];

int lowbit(int x){
    return x&(-x);
}
void update(int x,int d){
    while(x<maxn){
        tree[x]+=d;
        x +=lowbit(x);
    }
}

int query(int x){
    int ans=0;
    while(x>0){
        ans += tree[x];
        x -= lowbit(x);
    }
    return ans;
}

int main(){
    int n;
    while(scanf("%d",&n)){
        if(n==0) return 0;
        memset(tree,0,sizeof tree);
        for(int i=1;i<=n;i++){
            int a;
            int b;
            scanf("%d%d",&a,&b);
            update(a,1);
            update(b+1,-1);
        }
        for(int i=1;i<n;i++){
            printf("%d ",query(i));            
        }
        printf("%d\n",query(n));
    }

    return 0;
}

区间修改+区间查询

前面我们说 区间修改+单点查询 和 单点修改+区间查询

那么区间修改+区间查询那？ 树状数组当然也可以实现只不过需要树状数组套用树状数组，这种情况就可以直接切换线段树了。

我们用两个树状数组维护即可。一个维护区间修改，一个维护区间查询

///树状数组
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5 + 7;


ll tree1[maxn], tree2[maxn];

ll lowbit(ll x)
{
    return x & (-x);
}
void update1(ll x, ll d)
{
    while(x < maxn)
    {
        tree1[x] += d;
        x += lowbit(x);
    }
}
void update2(ll x, ll d)
{
    while(x < maxn)
    {
        tree2[x] += d;
        x += lowbit(x);
    }
}

ll query1(ll x)
{
    ll ans = 0;
    while(x > 0)
    {
        ans += tree1[x];
        x -= lowbit(x);
    }
    return ans;
}
ll query2(ll x)
{
    ll ans = 0;
    while(x > 0)
    {
        ans += tree2[x];
        x -= lowbit(x);
    }
    return ans;
}

int main()
{
    ll n, m;
    scanf("%lld%lld", &n, &m);
    ll top = 0;
    for(int i = 1; i <= n; i++)
    {
        ll a;
        scanf("%lld", &a);
        update1(i, a - top);
        update2(i, (a - top) * (i - 1));
        top = a;
    }
    while(m--)
    {
        ll q, l, r, d;
        scanf("%lld", &q);
        if(q == 1)
        {
            scanf("%lld%lld%lld", &l, &r, &d);
            update1(l, d);
            update1(r + 1, -d);
            update2(l, d * (l - 1));
            update2(r + 1, -d * r);
        }
        else
        {
            scanf("%lld%lld", &l, &r);
            printf("%lld\n", r * query1(r) - query2(r) - (l - 1)*query1(l - 1) + query2(l - 1));
        }
    }

    return 0;
}

二维区间修改+区间查询

P4514 上帝造题的七分钟
可以去做做。

///二位区间修改+区间查询
#include <bits/stdc++.h>
using namespace std;
const int maxn = 2050 + 7;
#define ll long long
#define x first
#define y second
#define pii pair<int ,int >
const int mod = 1e9 + 7;

int tree1[maxn][maxn], tree2[maxn][maxn], tree3[maxn][maxn], tree4[maxn][maxn];
int n, m;
int lowbit(int x)
{
    return x & (-x);
}
void update(int x, int y, int d)
{
    for(int i = x; i <= n; i += lowbit(i))
    {
        for(int j = y; j <= m; j += lowbit(j))
        {
            tree1[i][j] += d;
            tree2[i][j] += d * x;
            tree3[i][j] += d * y;
            tree4[i][j] += d * x * y;
        }
    }
}

int query(int x, int y)
{
    int ans = 0;
    for(int i = x; i > 0; i -= lowbit(i))
    {
        for(int j = y; j > 0; j -= lowbit(j))
        {
            ans += (x + 1) * (y + 1) * tree1[i][j] - (y + 1) * tree2[i][j] - (x + 1) * tree3[i][j] + tree4[i][j];
        }
    }
    return ans;
}


void solve()
{
    char ch[2];    
    scanf("%s",ch);
    scanf("%d%d",  &n, &m);

    int a, b, c, d;
    while(~scanf("%s",ch))
    {
        scanf("%d%d%d%d", &a, &b, &c, &d);
        if(ch[0] == 'L')
        {
            ll e;
            scanf("%d", &e);
            update(a, b, e);
            update(c + 1, d + 1, e);
            update(a, d + 1, -e);
            update(c + 1, b, -e);
        }
        else
        {
            printf("%d\n", query(c,d)+query(a-1,b-1)-query(a-1,d)-query(c,b-1));
        }
    }
}

int main()
{
    int t = 1;
    //scanf("%d", &t);
    while(t--)
    {
        solve();
    }
    return 0;
}

逆序对问题：

///二位区间修改+区间查询
#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e5 + 7;
#define ll long long
#define x first
#define y second
#define pii pair<int ,int >
const int mod = 1e9 + 7;

int n;
int tree[maxn],b[maxn];
int lowbit(int x){
    return x&(-x);
}
void upadte(int pos,int x){
    while(pos<=n){
        tree[pos]+=x;
        pos += lowbit(pos);
    }
}

ll query(int x){
    ll ans=0;
    while(x>0){
        ans+=tree[x];
        x -= lowbit(x);
    }
    return ans;
}
struct point{
  ll num,val;
}a[maxn];
bool cmp(point x,point y){
    if(x.val == y.val)   return x.num < y.num;  //注意：相等的情况，先出现标记更小
    return x.val < y.val;
}
void solve()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i].val);
        a[i].num = i;
    }
    sort(a+1,a+1+n,cmp);
    for (int i = 1 ;i <= n; i ++){
        b[a[i].num] = i;
    }
    ll ans=0;
    for(int i=n;i>=1;i--){
        upadte(b[i],1);
        ans+=query(b[i]-1);
    }
    printf("%lld\n",ans);
}

int main()
{
    int t = 1;
    //scanf("%d", &t);
    while(t--)
    {
        solve();
    }
    return 0;
}

树状数组维护区间最值

I Hate It(区间最大值)

///树状数组维护区间最值问题
#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e5 + 7;
#define ll long long
#define x first
#define y second
#define pii pair<int ,int >
const int mod = 1e9 + 7;

int n,m;
int tree[maxn], a[maxn];
int lowbit(int x)
{
    return x & (-x);
}
void upadte(int x, int d)
{
    while(x <= n)
    {
        tree[x] = d;
        for(int i = 1; i < lowbit(x); i <<= 1)
        {
            tree[x] = max(tree[x], tree[x - i]);
        }
        x += lowbit(x);
    }
}

int  query(int l, int r)
{
    int ans;
    while(l<=r){
        ans = max(ans,a[r]);
        r--;
        while (r-l>=lowbit(r)){
            ans =max(ans,tree[r]);
            r-=lowbit(r);
        }
    }
    return ans;
}
void solve()
{
    for(int i=1;i<=n;i++){        
        scanf("%d",&a[i]);
        upadte(i,a[i]);
    }
    while(m--){
        char ch[2];
        int l,r;
        scanf("%s",ch);
        scanf("%d%d",&l,&r);
        if(ch[0]=='Q'){
            printf("%d\n",query(l,r));
        }else {
            a[l]=r;
            upadte(l,r);
        }
    }
}

int main()
{
    ///int t = 1;
    //scanf("%d", &t);
    while(~scanf("%d%d",&n,&m))
    {
        solve();
    }
    return 0;
}