zoukankan      html  css  js  c++  java
  • 数据结构-树状数组-区间修改单点求值-1750. 区间加和求值

    2020-04-03 12:35:23

    问题描述:

    已知一个数列,你需要进行下面两种操作:
    1.将一个区间每一个数加上x
    2.求出某一个数的值
    输入:
    原数组为A。为了方便,A[0]为0.实际数列从A[1]开始。
    操作通过4元组给出。
    对于每个4元组(a,b,c,d):
    如果a=0 要求A[b]-A[c]区间的值都增加d(修改)。
    如果a=1 要求得到A[b]的值。其中c,d不起作用(查询)。

    输出:
    为了减少输出数据量。将操作2(询问)的所有值异或(^) 后返回。

    样例

    样例 1:

    输入:[0,1,2,3,4],[[1,1,0,0],[0,1,2,1],[1,2,0,0]]
    输出:2
    解释:
    第一个操作返回A [1] = 1
    第二个操作改变A为 [0,2,3,3,4]
    第三个操作返回A [1] = 3
    所以 1 ^ 3 = 2

    样例 2:

    输入:[0,1],[[1,1,0,0]]
    输出:1
    解释:第一个操作返回A [1] = 1,答案为 1。

    注意事项

    • 数列长度<=10000
    • 操作数<=50000

    问题求解:

    树状数组典型应用是单点更新,区间求和;如果要求解区间更新,单点求解的问题可以修改树状数组里存储的数值在O(logn)求解得到。

    a = [a1, a2, ... ,an]

    b = [a1, a2 - a1, .... , an - an-1]

    我们在树状数组中存储差分数组,这样单点查询直接query(i)即可;对于区间更新,只需要更新update(l, delta), update(r + 1, delta)。理由如下:

    a' = [a1, a2 + d, a3 + d,a4,...,an]

    b' = [a1, a2 - a1 + d,a3 - a2,a4 - a3 - d,a5 - a4,...,an - an-1]

        int[] bit;
        public long intervalsAddAndGetValue(int[] A, int[][] operations) {
            int n = A.length;
            bit = new int[n];
            for (int i = 1; i < n; i++) {
                update(i, A[i]);
                update(i + 1, -A[i]);
            }
            long res = 0;
            for (int[] op : operations) {
                if (op[0] == 0) {
                    update(op[1], op[3]);
                    update(op[2] + 1, -op[3]);
                }
                else {
                    res ^= query(op[1]);
                }
            }
            return res;
        }
        
        private void update(int idx, int delta) {
            for (int i = idx; i < bit.length; i += (i & -i)) {
                bit[i] += delta;
            }
        }
        
        private int query(int idx) {
            int res = 0;
            for (int i = idx; i > 0; i -= (i & -i)) {
                res += bit[i];
            }
            return res;
        }
    

      

  • 相关阅读:
    16.5 函数对象
    16.4.7 无序关联容器(C++11)
    16.4.6 关联容器
    16.4.5 容器种类(外1:7种序列容器类型)
    16.4.5 容器种类(下:序列)
    # SpringBoot + Spring AMQP 整合 RabbitMQ
    RabbitMQ 消息模型
    RabbitMQ Docker 单机与集群部署
    RabbitMQ 核心概念入门
    MQ消息中间件 + JMS + AMQP 理论知识
  • 原文地址:https://www.cnblogs.com/hyserendipity/p/12625785.html
Copyright © 2011-2022 走看看