【洛谷】P2068 统计和

题目地址:

https://www.luogu.com.cn/problem/P2068

题目描述:
给定一个长度为 n ( n ≤ 100000 ) n(n\leq 100000) n(n100000),初始值都为 0 0 0的序列, x ( x ≤ 100000 ) x(x\leq 100000) x(x100000)次的修改某些位置上的数字,每次加上一个数,然后提出 y ( y ≤ 100000 ) y(y\leq 100000) y(y100000)个问题,求每段区间的和。

输入格式:
第一行 1 1 1个整数,表示序列的长度 n n n
第二行 1 1 1个整数,表示操作的次数 w w w
后面依次是 w w w行,分别表示加入和询问操作。
其中,加入用x表示,询问用y表示。
x x x的格式为x a b表示在序列上第 a a a个数加上 b b b。保证 1 ≤ a ≤ n 1 \leq a \leq n 1an 1 ≤ b ≤ 1 0 9 1 \leq b \leq 10^9 1b109
y y y的格式为y a b表示询问 a a a b b b区间的加和。保证 1 ≤ a ≤ b ≤ n 1 \leq a \leq b \leq n 1abn

输出格式:
每行一个正整数,分别是每次询问的结果。

可以用线段树。代码如下:

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int n, m;
struct Node {
  int l, r;
  long sum;
} tr[N << 2];

void pushup(int u) {
  tr[u].sum = tr[u << 1].sum + tr[u << 1 | 1].sum;
}

void build(int u, int l, int r) {
  tr[u] = {l, r, 0};
  if (l == r) return;

  int mid = l + r >> 1;
  build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
}

void add(int u, int k, long x) {
  if (k <= tr[u].l && tr[u].r <= k) {
    tr[u].sum += x;
    return;
  }

  int mid = tr[u].l + tr[u].r >> 1;
  if (k <= mid) add(u << 1, k, x);
  else add(u << 1 | 1, k, x);
  pushup(u);
}

long query(int u, int l, int r) {
  if (l <= tr[u].l && tr[u].r <= r) return tr[u].sum;
  int mid = tr[u].l + tr[u].r >> 1;
  long res = 0;
  if (l <= mid) res += query(u << 1, l, r);
  if (r > mid) res += query(u << 1 | 1, l, r);
  return res;
}

int main() {
  scanf("%d%d", &n, &m);
  build(1, 1, n);
  while (m--) {
    char ch;
    cin >> ch;
    if (ch == 'x') {
      int k, x;
      scanf("%d%d", &k, &x);
      add(1, k, x);
    } else {
      int l, r;
      scanf("%d%d", &l, &r);
      printf("%ld\n", query(1, l, r));
    }
  }
}

每次操作时间复杂度 O ( log ⁡ n ) O(\log n) O(logn),空间 O ( n ) O(n) O(n)

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值