线段树特训

此处只讲一个,既有区间赋值操作,又有区间加值操作的线段树编写

其需要创建两种懒节点,并需要设置优先级

例题1:【模板】线段树 1 - 洛谷

示例代码:

#include <bits/stdc++.h>
#define int long long
using namespace std;

const int N = 1e5 + 10;
int n, m;
int d[4 * N], a[N], b[4 * N];

void build(int s, int t, int p) {
  if (s == t) {
    d[p] = a[s];
    return;
  }
  int m = (s + t) / 2;
  build(s, m, 2 * p);
  build(m + 1, t, 2 * p + 1);
  d[p] = d[2 * p] + d[2 * p + 1];
}

void update(int l, int r, int c, int s, int t, int p) {
  if (l <= s && t <= r) {
    d[p] += (t - s + 1) * c, b[p] += c;  // 如果区间被包含了,直接得出答案
    return;
  }
  int m = s + ((t - s) >> 1);
  if (b[p])
    d[2 * p] += b[p] * (m - s + 1), d[2 * p + 1] += b[p] * (t - m),
        b[2 * p] += b[p], b[2 * p + 1] += b[p];
  b[p] = 0;
  if (l <= m)
    update(l, r, c, s, m, 2 * p);  // 本行和下面的一行用来更新p*2和p*2+1的节点
  if (r > m) update(l, r, c, m + 1, t, 2 * p + 1);
  d[p] = d[p << 1] + d[(p << 1) | 1];  // 计算该节点区间和
}

int getsum(int l, int r, int s, int t, int p) {
  if (l <= s && t <= r) return d[p];
  int m = (s + t) / 2;
  if (b[p]) {
    d[2 * p] += b[p] * (m - s + 1);
    d[2 * p + 1] += b[p] * (t - m);
    b[2 * p] += b[p];
    b[2 * p + 1] += b[p];
    b[p] = 0;
  }
  int sum = 0;
  if (l <= m) sum += getsum(l, r, s, m, 2 * p);
  if (r > m) sum += getsum(l, r, m + 1, t, 2 * p + 1);
  return sum;
}

signed main() {
  cin >> n >> m;

  for (int i = 1; i <= n; i++) cin >> a[i];

  build(1, n, 1);

  while (m--) {
    int t, x, y, k;
    cin >> t >> x >> y;
    if (t == 1) {
      cin >> k;

      update(x, y, k, 1, n, 1);
      continue;
    }

    cout << getsum(x, y, 1, n, 1) << endl;
    ;
  }
}

例题2:【模板】线段树 2 - 洛谷

例题3:线段树的区间修改 - HihoCoder 1078 - Virtual Judge

#include <bits/stdc++.h>
#define int long long
using namespace std;

const int N = 1e5 + 10;
int n, m;
int d[4 * N], a[N], b[4 * N];

void build(int s, int t, int p) {
  if (s == t) {
    d[p] = a[s];
    return;
  }
  int m = (s + t) / 2;
  build(s, m, 2 * p);
  build(m + 1, t, 2 * p + 1);
  d[p] = d[2 * p] + d[2 * p + 1];
}

void update(int l, int r, int c, int s, int t, int p) {
  if (l <= s && t <= r) {
    d[p] += (t - s + 1) * c, b[p] = c;  // 如果区间被包含了,直接得出答案
    return;
  }
  int m = s + ((t - s) >> 1);
  if (b[p]) {
    d[2 * p] = b[p] * (m - s + 1), d[2 * p + 1] = b[p] * (t - m),
          b[2 * p] = b[p], b[2 * p + 1] = b[p];
    b[p] = 0;
  }

  if (l <= m)
    update(l, r, c, s, m, 2 * p);  // 本行和下面的一行用来更新p*2和p*2+1的节点
  if (r > m) update(l, r, c, m + 1, t, 2 * p + 1);
  d[p] = d[2 * p] + d[2 * p + 1];  // 计算该节点区间和
}

int getsum(int l, int r, int s, int t, int p) {
  if (l <= s && t <= r) return d[p];
  int m = (s + t) / 2;
  if (b[p]) {
    d[2 * p] = b[p] * (m - s + 1);
    d[2 * p + 1] = b[p] * (t - m);
    b[2 * p] = b[p];
    b[2 * p + 1] = b[p];
    b[p] = 0;
  }
  int sum = 0;
  if (l <= m) sum += getsum(l, r, s, m, 2 * p);
  if (r > m) sum += getsum(l, r, m + 1, t, 2 * p + 1);
  return sum;
}

signed main() {
  cin >> n >> m;

  for (int i = 1; i <= n; i++) cin >> a[i];

  build(1, n, 1);

  while (m--) {
    int t, x, y, k;
    cin >> t >> x >> y;
    if (t == 1) {
      cin >> k;

      update(x, y, k, 1, n, 1);
      continue;
    }

    cout << getsum(x, y, 1, n, 1) << endl;
    ;
  }
}

例题4:Problem - 6356

  • 9
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值