线段树 + 树状数组 + ST表 模板

12 篇文章 0 订阅

线段树

区间修改+区间求和 logN

const int N = 1e5 + 5;
int a[N];
namespace SEG
{
    struct tag
    {
        long long sum;
        long long lazy;
    } c[N << 2];
    void pushDown(int k, int l, int r)
    {
        int mid = (l + r) >> 1;

        c[k << 1].sum += c[k].lazy*(mid - l + 1);
        c[k << 1 | 1].sum += c[k].lazy*(r - mid);

        c[k << 1].lazy += c[k].lazy;
        c[k << 1 | 1].lazy += c[k].lazy;

        c[k].lazy = 0;
    }
    void pushUp(int k)
    {
        c[k].sum = c[k << 1].sum + c[k << 1 | 1].sum;
    }
    // l, r是总区间 left, right是查询区间
    long long query(int k, int l, int r, int left, int right)
    {
        if (left <= l && r <= right)    return c[k].sum;

        pushDown(k, l, r);

        long long ans = 0;

        int mid = (l + r) >> 1;
        if (left <= mid)    ans += query(k << 1, l, mid, left, right);
        if (mid < right)    ans += query(k << 1 | 1, mid + 1, r, left, right);

        return ans;
    }
    void build(int k, int l, int r)
    {
        c[k].lazy = 0;
        if (l == r)
        {
            c[k].sum = a[l];  //a是原始数据数组
            return;
        }
        int mid = (l + r) >> 1;

        build(k << 1, l, mid);
        build(k << 1 | 1, mid + 1, r);

        pushUp(k);
    }
    // l, r是总区间 left, right是查询区间,k是起始节点编号,val是更新的值
    void update(int k, int l, int r, int left, int right, int val)
    {
        if (left <= l && r <= right)
        {
            c[k].lazy += val;
            c[k].sum += val * (r - l + 1);
            return;
        }

        pushDown(k, l, r);

        int mid = (l + r) >> 1;
        if (left <= mid)    update(k << 1, l, mid, left, right, val);
        if (mid < right)    update(k << 1 | 1, mid + 1, r, left, right, val);

        pushUp(k);
    }
}

树状数组

区间求和+单点修改 logN

const int N = 1e5 + 5;
namespace BIT
{
    long long c[N];
    inline int lowbit(int x) { return x & (~x + 1); }
    // 查询[1, x]的区间和
    inline long long sum(int x)
    {
        long long res = 0;
        for (int i = x; i; i -= lowbit(i))  res += c[i];
        return res;
    }
    // 在x处加上val
    inline void add(int x, long long val)
    {
        for (int i = x; i < N; i += lowbit(i))  c[i] += val;
    }
}

ST表

离线查询区间最值 构造NlogN 查询1

#include <cstring>
const int N = 500005;
int a[N];
inline int max(int x, int y) { return x > y ? x : y; }
inline int min(int x, int y) { return x > y ? y : x; }
namespace ST
{
    int log[N];
    int stmin[N][30], stmax[N][30];
    // 快速查询log[n]
    void pre()
    {
        log[1] = 0;
        for (int i = 2; i < N; ++i)
        {
            log[i] = log[i - 1];
            if (i == (1 << (log[i] + 1)))   log[i]++;
        }
    }
    // 构造ST表
    void init()
    {
        memset(stmin, 0, sizeof(stmin));
        memset(stmax, 0, sizeof(stmax));
        for (int i = 1; i <= N; i++)
            stmax[i][0] = stmin[i][0] = a[i];
        for (int j = 1; j <= log[N]; j++)
            for (int i = 1; i + (1 << j) - 1 <= N; i++)
            {
                stmax[i][j] = max(stmax[i][j - 1], stmax[i + (1 << (j - 1))][j - 1]);
                stmin[i][j] = min(stmin[i][j - 1], stmin[i + (1 << (j - 1))][j - 1]);
            }
    }
    // 查询区间[l, r]的最大值
    int queryMax(int l, int r)
    {
        int k = log[r - l + 1];
        return max(stmax[l][k], stmax[r - (1 << k) + 1][k]);
    }
    // 查询区间[l, r]的最小值
    int queryMin(int l, int r)
    {
        int k = log[r - l + 1];
        return min(stmin[l][k], stmin[r - (1 << k) + 1][k]);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
线段树树状数组都是用来解决区间相关问题的数据结构。 线段树是一种二叉树形式的数据结构,用于解决区间查询问题。每个节点示一个区间,根节点示整个区间,通过对区间进行适当的划分,将原问题划分为子问题,递归地构建线段树线段树的叶子节点示原始数组的单个元素,而其他节点示其子区间的一些统计信息,如和、最大值、最小值等。通过适当的操作,可以在O(logN)的时间内查询区间的统计信息,也可以在O(logN)的时间内更新一个元素或一个区间的值。 树状数组是一种实现类似累加的数据结构,用于解决前缀查询问题。树状数组的底层数据结构是一个数组,通过对数组的某些位置进行增加或查询操作,可以在O(logN)的时间内得到累加值。数组的索引和实际数值之间存在一种特殊的关系,即某个位置的累加值等于该位置的二进制示中最低位的连续1的个数。树状数组的区间查询通过将原始数组转换为差分数组来实现,将查询问题转换为若干个单点查询。 线段树树状数组在解决问题时都具有一些特定的优势和适用场景。线段树适用于一些需要频繁修改和查询区间统计信息的问题,如区间最值、区间和等。而树状数组适用于一些需要频繁查询前缀和的问题,如求逆序对的数量或统计小于某个数的元素个数等。根据具体的问题需要,我们可以选择合适的数据结构来解决和优化计算效率。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值