[Luogu] P1438 无聊的数列 | 线段树简单题

简介: 题目背景无聊的 YYB 总喜欢搞出一些正常人无法搞出的东西。有一天,无聊的 YYB 想出了一道无聊的题:无聊的数列。。。

题目背景


无聊的 YYB 总喜欢搞出一些正常人无法搞出的东西。有一天,无聊的 YYB 想出了一道无聊的题:无聊的数列。。。(K峰:这题不是傻X题吗)


题目描述


维护一个数列 a i a_ia

i

 ,支持两种操作:


1 l r K D:给出一个长度等于 r − l + 1 r-l+1r−l+1 的 等差数列,首项为 K KK,公差为 D DD,并将它对应加到 [ l , r ] [l,r][l,r] 范围中的每一个数上。即:令 al=al+K,al+1=al+1+K+Dar=ar+K+(rl)×D

2 p:询问序列的第 p 个数的值 ap


输入格式


第一行两个整数数 n , m 表示数列长度和操作个数。

第二行 n  个整数,第 i  个数表示ai


接下来的 m mm 行,每行先输入一个整数 o p t optopt

若opt=1 则再输入四个整数 l   r   K   D

若 opt=2 则再输入一个整数 p


输出格式


对于每个询问,一行一个整数表示答案。


输入输出样例


输入

5 2
1 2 3 4 5
1 2 4 1 2
2 3


输出

6


说明/提示


数据规模与约定

对于100% 数据,0 ≤ n , m ≤ 1 0 5 , − 200 ≤ a i , K , D ≤ 200 1lrn,1pn


根据差分很容易就可以想到:

在对区间[ l , r ] 加上一个首项为K ,公差为D 的等差数列之后,我们可以根据差分得到这样的式子:

a[l]=a[l]+K

a[i]=a[i]+D 其中 i ∈ [ l + 1 , r ] i \in [l+1,r]i∈[l+1,r]

a[r+1]=a[r+1]−K−(r−(l+1)+1)∗D 其中,R + 1 ≤ n

对于要查询的答案,应该为a[p] + ∑ i = 1 p s u m [ i ] \sum_{i=1}^p sum[i]∑

i=1

p

sum[i]


ac_code:


#define mid ((l + r) >> 1)
int n, m;
ll a[maxn << 2];
ll sum[maxn << 2];
ll lazy[maxn << 2];
void PushUp(int rt) {
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}
void PushDown(int rt, ll len) {
    if (lazy[rt]) {
        lazy[rt << 1] += lazy[rt];
        lazy[rt << 1 | 1] += lazy[rt];
        sum[rt << 1] += (len - (len >> 1)) * lazy[rt];
        sum[rt << 1 | 1] += (len >> 1) * lazy[rt];
        lazy[rt] = 0;
    }
}
void Update(int rt, int l, int r, int L, int R, ll val) {
    if (L <= l && r <= R) {
        sum[rt] += (r - l + 1) * val;
        lazy[rt] += val;
        return;
    }
    PushDown(rt, r - l + 1);
    int md = mid;
    if (md >= L) Update(rt << 1, l, md, L, R, val);
    if (md < R) Update(rt << 1 | 1, mid + 1, r, L, R, val);
    PushUp(rt);
}
ll Query(int rt, int l, int r, int L, int R) {
    if (L <= l && r <= R) return sum[rt];
    PushDown(rt, r - l + 1);
    int md = mid;
    ll ret = 0;
    if (md >= L) ret += Query(rt << 1, l, md, L, R);
    if (md < R) ret += Query(rt << 1 | 1, md + 1, r, L, R);
    return ret;
}
int main() {
    n = read, m = read;
    for (int i = 1; i <= n; i++) a[i] = read;
    ll L, R, K, D, op;
    int p;
    // puts("ok");
    while (m--) {
        op = read;
        if (op == 1) {
            L = read, R = read, K = read, D = read;
            Update(1, 1, n, L, L, K);
            if (L < R) Update(1, 1, n, L + 1, R, D);
            ll tot = R - L;
            if (R != n) Update(1, 1, n, R + 1, R + 1, -1 * (K + (tot * D)));
        } else {
            p      = read;
            ll ans = a[p] + Query(1, 1, n, 1, p);
            printf("%lld\n", ans);
        }
    }
    return 0;
}
/**
**/


目录
相关文章
|
7月前
|
算法 测试技术 C++
C++算法:美丽塔O(n)解法单调栈
C++算法:美丽塔O(n)解法单调栈
|
12天前
|
C语言
大衍数列(蓝桥杯)
大衍数列(蓝桥杯)
|
2月前
|
存储
每日一题啦(● ̄(エ) ̄●)(尼克切斯定理,等差数列)
每日一题啦(● ̄(エ) ̄●)(尼克切斯定理,等差数列)
10 0
【剑指offer】-扑克牌顺子-44/67
【剑指offer】-扑克牌顺子-44/67
|
7月前
|
机器学习/深度学习 存储 C++
[蓝桥杯] 树状数组与线段树问题(C/C++)
[蓝桥杯] 树状数组与线段树问题(C/C++)
53 0
|
8月前
|
机器学习/深度学习 算法
蓝桥杯:递推算法 递归算法 例题:斐波纳契数列
蓝桥杯:递推算法 递归算法 例题:斐波纳契数列
41 0
|
11月前
|
算法 测试技术 Python
第十二届蓝桥杯《杨辉三角》-二分法
第十二届蓝桥杯《杨辉三角》-二分法
61 0
|
11月前
|
机器学习/深度学习 人工智能 移动开发
|
11月前
|
人工智能 BI
|
11月前
|
机器学习/深度学习 人工智能 移动开发