首页 > 其他分享 >NC50454 A Simple Problem with Integers

NC50454 A Simple Problem with Integers

时间:2023-04-27 23:01:56浏览次数:49  
标签:rt Integers NC50454 le return Simple sum update int

题目链接

题目

题目描述

给定数列 \(a[1],a[2], \dots,a[n]\) ,你需要依次进行q个操作,操作有两类:

1 l r x:给定l,r,x,对于所有 \(i \in[l,r]\) ,将a[i]加上x(换言之,将 \(a[l],a[l+1], \dots,a[r]\) 分别加上x);

2 l r:给定l,r,求 \(\sum_{i=l}^ra[i]\) 的值(换言之,求 \(a[l]+a[l+1]+ \dots+a[r]\) 的值)。

输入描述

第一行包含2个正整数n,q,表示数列长度和询问个数。保证 \(1 \le n,q \le10^6\) 。
第二行n个整数 \(a[1],a[2], \dots,a[n]\) ,表示初始数列。保证 \(|a[i]| \le10^6\) 。
接下来q行,每行一个操作,为以下两种之一:
1 l r x:对于所有 \(i \in[l,r]\) ,将a[i]加上x;
2 l r:输出 \(\sum_{i=l}^ra[i]\) 的值。保证 \(1 \le l \le r \le n\) , \(|x| \le10^6\) 。

输出描述

对于每个2lr操作,输出一行,每行有一个整数,表示所求的结果。

示例1

输入

5 10
2 6 6 1 1
2 1 4
1 2 5 10
2 1 3
2 2 3
1 2 2 8
1 2 3 7
1 4 4 10
2 1 2
1 4 5 6
2 3 4

输出

15
34
32
33
50

备注

对于所有数据,\(1 \le n,q \le10^6\) , \(|a[i]| \le10^6\) , \(1 \le l \le r \le n\) , \(|x| \le10^6\) 。

题解

方法一

知识点:线段树。

可以用线段树,不过因为线段树空间常数是四倍,容易炸空间,但还是有概率能过。

时间复杂度 \(O((n+q)\log n)\)

空间复杂度 \(O(n)\)

方法二

知识点:树状数组。

本题正解,是用树状数组维护区间和、区间加,是个板子题。

设原数组为 \(a\) ,差分数组为 \(d\) ,有如下公式:

\[\begin{aligned} sum_{1,x} &= \sum_{i=1}^{x} a_i \\ &= \sum_{i=1}^{x} \sum_{j=1}^{i} d_j \\ &= \sum_{j=1}^{x} \sum_{i=j}^{x} d_j \\ &= \sum_{j=1}^{x} (x-j+1)d_j \\ &= (x+1) \sum_{j=1}^{x} d_j - \sum_{j=1}^{x} jd_j \\ \end{aligned} \]

因此只需要维护 \(d_j\) 和 \(jd_j\) 的前缀和即可。同时,在差分意义下,区间加转化为两次单点加。

时间复杂度 \(O((n+q)\log n)\)

空间复杂度 \(O(n)\)

代码

方法一

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

struct T {
    int len;
    ll sum;
    static T e() { return { 0,0 }; }
    friend T operator+(const T &a, const T &b) { return { a.len + b.len, a.sum + b.sum }; }
};
struct F {
    ll add;
    static F e() { return { 0 }; }
    T operator()(const T &x) { return { x.len,x.sum + add * x.len }; }
    F operator()(const F &g) { return { g.add + add }; }
};
template<class T, class F>
class SegmentTreeLazy {
    int n;
    vector<T> node;
    vector<F> lazy;

    void push_down(int rt) {
        node[rt << 1] = lazy[rt](node[rt << 1]);
        lazy[rt << 1] = lazy[rt](lazy[rt << 1]);
        node[rt << 1 | 1] = lazy[rt](node[rt << 1 | 1]);
        lazy[rt << 1 | 1] = lazy[rt](lazy[rt << 1 | 1]);
        lazy[rt] = F::e();
    }

    void update(int rt, int l, int r, int x, int y, F f) {
        if (r < x || y < l) return;
        if (x <= l && r <= y) return node[rt] = f(node[rt]), lazy[rt] = f(lazy[rt]), void();
        push_down(rt);
        int mid = l + r >> 1;
        update(rt << 1, l, mid, x, y, f);
        update(rt << 1 | 1, mid + 1, r, x, y, f);
        node[rt] = node[rt << 1] + node[rt << 1 | 1];
    }

    T query(int rt, int l, int r, int x, int y) {
        if (r < x || y < l)return T::e();
        if (x <= l && r <= y) return node[rt];
        push_down(rt);
        int mid = l + r >> 1;
        return query(rt << 1, l, mid, x, y) + query(rt << 1 | 1, mid + 1, r, x, y);
    }

public:
    SegmentTreeLazy(int _n = 0) { init(_n); }
    SegmentTreeLazy(const vector<T> &src) { init(src); }

    void init(int _n) {
        n = _n;
        node.assign(n << 2, T::e());
        lazy.assign(n << 2, F::e());
    }
    void init(const vector<T> &src) {
        assert(src.size());
        init(src.size() - 1);
        function<void(int, int, int)> build = [&](int rt, int l, int r) {
            if (l == r) return node[rt] = src[l], void();
            int mid = l + r >> 1;
            build(rt << 1, l, mid);
            build(rt << 1 | 1, mid + 1, r);
            node[rt] = node[rt << 1] + node[rt << 1 | 1];
        };
        build(1, 1, n);
    }

    void update(int x, int y, F f) { update(1, 1, n, x, y, f); }

    T query(int x, int y) { return query(1, 1, n, x, y); }
};

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n, q;
    cin >> n >> q;
    vector<T> a(n + 1);
    for (int i = 1;i <= n;i++) {
        int x;
        cin >> x;
        a[i] = { 1,x };
    }
    SegmentTreeLazy<T, F> sgt(a);
    while (q--) {
        int op, l, r;
        cin >> op >> l >> r;
        if (op == 1) {
            int x;
            cin >> x;
            sgt.update(l, r, { x });
        }
        else cout << sgt.query(l, r).sum << '\n';
    }
    return 0;
}

方法二

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

struct T {
    ll sum;
    static T e() { return { 0 }; }
    T &operator+=(const T &x) { return sum += x.sum, *this; }
};

template<class T>
class Fenwick {
    int n;
    vector<T> node;

public:
    Fenwick(int _n = 0) { init(_n); }

    void init(int _n) {
        n = _n;
        node.assign(n + 1, T::e());
    }

    void update(int x, T val) { for (int i = x;i <= n;i += i & -i) node[i] += val; }

    T query(int x) {
        T ans = T::e();
        for (int i = x;i >= 1;i -= i & -i) ans += node[i];
        return ans;
    }
};

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n, q;
    cin >> n >> q;
    Fenwick<T> d(n + 1), di(n + 1);
    auto update = [&](int l, int r, T val) {
        d.update(l, { val.sum }), d.update(r + 1, { -val.sum });
        di.update(l, { l * val.sum }), di.update(r + 1, { -(r + 1) * val.sum });
    };
    auto query = [&](int l, int r) {
        return ((r + 1) * d.query(r).sum - di.query(r).sum) -
            (l * d.query(l - 1).sum - di.query(l - 1).sum);
    };
    for (int i = 1;i <= n;i++) {
        int x;
        cin >> x;
        update(i, i, { x });
    }
    while (q--) {
        int op, l, r;
        cin >> op >> l >> r;
        if (op == 1) {
            int x;
            cin >> x;
            update(l, r, { x });
        }
        else cout << query(l, r) << '\n';
    }
    return 0;
}

标签:rt,Integers,NC50454,le,return,Simple,sum,update,int
From: https://www.cnblogs.com/BlankYang/p/17360481.html

相关文章

  • NC51100 A Simple Problem with Integers
    题目链接题目题目描述YouhaveNintegers,\(A_1,A_2,...,A_N\).Youneedtodealwithtwokindsofoperations.Onetypeofoperationistoaddsomegivennumbertoeachnumberinagiveninterval.Theotheristoaskforthesumofnumbersinagivenint......
  • java中使用RedisTemplate读取数据异常 Missing type id when trying to resolve subt
    java中使用RedisTemplate读取数据异常Missingtypeidwhentryingtoresolvesubtypeof[simpletype,classjava.lang.Object]:missingtypeidproperty'@class'at[Source:(byte[])"报错:Causedby:com.fasterxml.jackson.databind.exc.InvalidTypeIdExcep......
  • drf-jwt、simplejwt的使用
    1.接口文档#前后端分离 -我们做后端,写接口-前端做前端,根据接口写app,pc,小程序-作为后端来讲,我们很清楚,比如登录接口/api/v1/login/---->post---->username,password编码方式json----》返回的格式{code:100,msg:登录成功}-后端人员,接口写完,一......
  • pip 安装库是报错ERROR: Cannot unpack file C:\Users\LX\AppData\Local\Temp\p
    使用pip安装python库的时候出现报错:ERROR:CannotunpackfileC:\Users\LX\AppData\Local\Temp\pip-unpack-apk_4xkw\simple(downloadedfromC:\Users\LX\AppData\Local\Temp\pip-req-build-htbv29co,content-type:text/html;charset=utf-8);cannotdetectarch......
  • uhttp luci simple-app
    uhttplucisimple-appseehttps://forum.openwrt.org/t/rest-api-supported-in-openwrt/17210/6 YouprobablydowantLuci,oratleasttotakealookathowLucidoesit.POSTdataisinthebodyoftherequest.Lucicreatesa ltn12compatiblesourcetorea......
  • MIT6.5830-2022 Lab 1: SimpleDB
    SimpleDB组成:classes:表示fields,tuples,tupleschemas。classes:作用于tuples的谓词和条件类。methods:在硬盘上存储关系(如heapfiles),处理并发控制和事务。classes:处理tuples的操作类(select,join,insert,delete等)。bufferpool:在内存中缓存活跃的......
  • 博客园SimpleMemory美化搬运工教程
    这不是我写的代码,但是,我是负责把写代码的大佬的代码分享出来。接下来,就开始吧。1.开通博客权限。点击左上角头像,然后点击左上角我的博客,如果你没有开通博客,它会让你开通,这个你只需要申请,因为是人工审核,耐心等几十分钟至几小时不等就可以被批准,开通理由随便写点正当的什么......
  • heatmapts_simple-heatmap的使用
    simpleheat的使用<scriptsetuplang="ts">import{SimpleHeat}from"simpleheat-ts";import*asdatfrom"dat.gui";letframe:number|null=null;constoCanvas=document.createElement("canvas");oCanva......
  • Django Admin SimpleUI 自定义列
    DjangoAdminSimpleUI的一些简单用法注意事项:自定义列会遍历所有取出来的数据,需注意先定义下模型#models.pyclassDog(models.Model):name=models.CharField(max_length=15,verbose_name='小狗名字')age=models.IntegerField(verbose_name='小狗年龄',defa......
  • django admin 使用SimpleUI自定义按钮弹窗框示例
    actions=['button_exeScript',]defbutton_exeScript(self,request,queryset):passbutton_exeScript.layer={#弹出层中的输入框配置#这里指定对话框的标题'title':'弹出层输入框',#提示信息......