校门外的树（线段树）

最新推荐文章于 2024-08-22 17:27:46 发布

念兮兮

最新推荐文章于 2024-08-22 17:27:46 发布

阅读量110

点赞数

分类专栏：《信息学奥赛一本通-提高篇》在线题库文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_52081267/article/details/129735091

版权

《信息学奥赛一本通-提高篇》在线题库专栏收录该内容

40 篇文章 0 订阅

订阅专栏

该文章展示了一个C++代码示例，使用树状数组数据结构来维护和查询括号序列中的平衡状态。程序处理两种操作：添加括号对和查询区间内括号是否平衡。lowbit函数用于计算二进制表示中最低位的1，add和query函数分别用于更新和查询树状数组。

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

原题链接

解题思路：用树状数组维护左右括号

C++代码

#include <bits/stdc++.h>
// #define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
using namespace std;
const int N = 5e4 + 10;
long long a[N], cl[N], cr[N];
int n, q;
int lowbit(int x)  // 返回末尾的1
{
    return x & -x;
}
void add(int x, int y, long long *c)
{
    for (int i = x; i <= N; i += lowbit(i))
    {
        c[i] += y;
    }
}
long long query(int x, long long *c)
{
    long long sum = 0;
    for (int i = x; i; i -= lowbit(i))
    {
        sum += c[i];
    }
    return sum;
}
signed main()
{
//     IOS;
    scanf("%d%d", &n, &q);
    for (int i = 1; i <= q; i ++ ) 
    {
        int op, l, r;
        scanf("%d%d%d", &op, &l, &r);
        if (op == 1)
        {
            add(l, 1, cl);
            add(r, 1, cr);
        }
        else 
        {
            cout << query(r, cl) - query(l - 1, cr) << endl;
        }
    }
    return 0;
}

参考博客