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

最新推荐文章于 2024-07-17 23:01:07 发布

算法下的星辰曲

最新推荐文章于 2024-07-17 23:01:07 发布

阅读量295

点赞数 13

分类专栏：蓝桥杯文章标签： c++ 开发语言

本文链接：https://blog.csdn.net/m0_74850682/article/details/135950845

版权

蓝桥杯专栏收录该内容

7 篇文章 0 订阅

订阅专栏

文章讲述了如何使用树状数组数据结构解决AcWing平台上的两个问题：动态求连续区间和和数星星。树状数组用于高效计算区间和，并提供了相应的代码实现和分析思路。

摘要由CSDN通过智能技术生成

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