首页 > 其他分享 >COGS 2632. [HZOI 2016] 数列操作d

COGS 2632. [HZOI 2016] 数列操作d

时间:2022-10-25 14:04:07浏览次数:82  
标签:COGS d% tree 2632 int add HZOI include mod


题目链接:​​传送门​​ 区间加等差数列

网上没怎么见有解释的

把等差数列看成一个三角形

mid从中间切开后

左儿子加的还是一个三角形

右儿子加的是一个梯形

其实也不能说是三角形

因为怎么样都会有一个首项在

也就是底下有一个小正方形

也就是这个样子↓

维护这个面积

(上底+下底)*高 / 2

COGS 2632. [HZOI 2016] 数列操作d_线段树


这样维护区间和就好了

这里的首项和公差是当做懒标记下放的

下放给右儿子的时候要注意首项是变的

如图(好像有点歪

COGS 2632. [HZOI 2016] 数列操作d_等差数列_02

首项变成了a+d*(m-l+1)
只不过
为什么多取模会错!

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <complex>
#include <algorithm>
#include <climits>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iomanip>
#define
#define

using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
struct node {
int l, r; int w, a, d;
}tree[A];
void build(int k, int l, int r) {
tree[k].l = l; tree[k].r = r;
if (l == r) {
tree[k].w = tree[k].a = tree[k].d = 0;
return;
}
int m = (l + r) >> 1;
build(k << 1, l, m);
build(k << 1 | 1, m + 1, r);
}
void add(int k, int l, int r, int a, int d) {
tree[k].a = (tree[k].a + a) % mod;
tree[k].d = (tree[k].d + d) % mod;
tree[k].w = (tree[k].w + (2LL * a + (r - l) * d) * (r - l + 1) / 2 % mod) % mod;
}
void down(int k) {
int m = (tree[k].l + tree[k].r) >> 1;
add(k << 1, tree[k].l, m, tree[k].a, tree[k].d);
add(k << 1 | 1, m + 1, tree[k].r, (tree[k].a + tree[k].d * 1LL * (m - tree[k].l + 1) % mod) % mod, tree[k].d);
tree[k].a = tree[k].d = 0;
}
void change(int k, int l, int r, int d) {
if (tree[k].l >= l and tree[k].r <= r) {
add(k, tree[k].l, tree[k].r, 1LL * (tree[k].l - l) * d % mod, d);
return;
}
if (tree[k].d) down(k);
int m = (tree[k].l + tree[k].r) >> 1;
if (l <= m) change(k << 1, l, r, d);
if (r > m) change(k << 1 | 1, l, r, d);
tree[k].w = (tree[k << 1].w + tree[k << 1 | 1].w) % mod;
}
int ask(int k, int l, int r) {
if (tree[k].l >= l and tree[k].r <= r) return tree[k].w;
if (tree[k].d) down(k);
int m = (tree[k].l + tree[k].r) >> 1; int ans = 0;
if (l <= m) ans = (ans + ask(k << 1, l, r)) % mod;
if (r > m) ans = (ans + ask(k << 1 | 1, l, r)) % mod;
return ans;
}
int n, m, opt, a, b; int c;

int main(int argc, char const *argv[]) {
freopen("segment.in", "r", stdin); freopen("segment.out", "w", stdout);
scanf("%d%d", &n, &m); build(1, 1, n);
for (int i = 1; i <= m; i++) {
scanf("%d", &opt);
if (!opt) {
scanf("%d%d", &a, &b);
printf("%d\n", ask(1, a, b));
}
else {
scanf("%d%d%d", &a, &b, &c);
change(1, a, b, c);
}
}
fclose(stdin); fclose(stdout);
return 0;
}


标签:COGS,d%,tree,2632,int,add,HZOI,include,mod
From: https://blog.51cto.com/lyle/5794666

相关文章