1264. 动态求连续区间和

最新推荐文章于 2024-03-22 18:51:19 发布

StephenZhen

最新推荐文章于 2024-03-22 18:51:19 发布

阅读量109

点赞数

分类专栏： java 文章标签： java leetcode 数据结构算法

本文链接：https://blog.csdn.net/z18607005331/article/details/105759620

版权

java 专栏收录该内容

26 篇文章 0 订阅

订阅专栏

题意

给定 n 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b]的连续和
输入样例：
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8

输出样例：
11
30
35
常规方法可能会出现超时的情况，因此采用树状的数组来存储这个和，时间满足；

常规方法：

import java.util.*;
public class Main{
    public static void main(String[] agrs){
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int m = in.nextInt();
        int []num = new int[n+1];
        for(int i = 1; i<=n; i++){
            num[i] = num[i-1] + in.nextInt();
        }
        for(int i = 0 ; i < m; i++){
            int [] cur = new int[3];
            cur[0] = in.nextInt();
            cur[1] = in.nextInt();
            cur[2] = in.nextInt();
            if(cur[0]==0){
                System.out.println(num[cur[2]]-num[cur[1]-1]);
            }else if(cur[0]==1){
                while(cur[1]<=n){
                    num[cur[1]]+=cur[2];
                    cur[1]++;
                }
            }
        }
    }
}

改进方法：


import java.util.*;
public class Main{
    static int n;
    static int m;
    static int []cur = new int[100100];

    public static void main(String[] agrs){
        Scanner in = new Scanner(System.in);
        n = in.nextInt();
        m = in.nextInt();
        int []num = new int[n+1];
        for(int i = 1; i<=n; i++){
            num[i] = in.nextInt();
            add(i,num[i]);
        }
        
        int []k = new int[3];
        for(int j = 0; j < m; j++){
            k[0] = in.nextInt();
            k[1] = in.nextInt();
            k[2] = in.nextInt();
            if(k[0]==0){
                System.out.println(summ(k[2])-summ(k[1]-1));
            }else{
                add(k[1],k[2]);
            }
        }
        
    }
    public static void add(int x,int y){
        for(int i = x; i <= n; i+=i&(-i)){
            cur[i]+=y;
        }
    }
    public static int summ(int x){
        int res=0;
        for(int i = x; i > 0; i-=i&(-i)){
            res+=cur[i];
        }
        return res;
    }
}