C++算法之树状数组与线段树

AcWing 1264. 动态求连续区间和 详细题解AcWing,题解,动态求连续区间和,https://www.acwing.com/solution/content/7526/

一、树状数组

1.AcWing 1264.动态求连续区间和 

分析思路

树状数组c[x]:表示（x-lowbit(x),x]区域的和

一个数改变同时也要改变其他位置的数组，下一个父节点是i+lowbit(i)

代码实现

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;
const int N=100010;
int a[N],s[N];
int n,m;

int lowbit(int x)
{
    return x & -x;
}
void add(int x,int y)
{
    for(int i=x;i<=n;i+=lowbit(i)) s[i]+=y;
}
int query(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=s[i];
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) add(i,a[i]);

    while(m--)
    {
        int k,a,b;
        scanf("%d%d%d",&k,&a,&b);
        if(k==0) printf("%d\n", query(b) - query(a - 1));
        else add(a,b);
    }
    return 0;
}

2.AcWing 1265. 数星星

分析思路

a[x]:用来表示横坐标是x的个数

sum(x):表示<x的个数

level[i]:表示各等级的个数

代码实现 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
const int N = 32010;
int n;
int a[N],level[N];

int lowbit(int x)//看二进制数后有多少个连续的0
{
    return x&-x;
}
void add(int x)
{
    for(int i=x;i<N;i+=lowbit(i)) a[i]++;
}
int sum(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=a[i];
    return res;
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        x++;
        level[sum(x)]++;
        add(x);
    }
    
    for(int i=0;i<n;i++) printf("%d\n",level[i]);
    
    return 0;
}