【蓝桥杯省赛学习题Java】大学里的树木

题目描述

教室外有 N 棵树（树的编号从 0∼N−1），根据不同的位置和树种，学校要对其上不同的药。

因为树的排列成线性，且非常长，我们可以将它们看作一条直线给他们编号。

对于树的药是成区间分布，比如 3∼5 号的树靠近下水道，所以他们要用驱蚊虫的药， 20∼26 号的树，他们排水不好，容易涝所以要给他们用点促进根系的药 ⋯诸如此类。

每种不同的药要花不同的钱。

现在已知共有 M 个这样的区间，并且给你每个区间花的钱，问最后这些树木要花多少药费。

输入描述

每组输入的第一行有两个整数 N和 M。N 代表马路的共计多少棵树，M 代表区间的数目，N 和 M 之间用一个空格隔开。

接下来的 M 行每行包含三个不同的整数，用一个空格隔开，分别表示一个区域的起始点 L 和终止点 R 的坐标，以及花费。

1≤L≤R≤N≤10^6,1≤M≤105，保证花费总和不超过 int 范围。

输出描述

输出包括一行，这一行只包含一个整数，所有的花费。

输入

500 3
150 300 4
100 200 20
470 471 19

输出

2662

源码实现

差分

public class 大学里的树木要打药 {
    static int n;//树的总数
    static int m;//区间数
    static int[] trees;
    static int res = 0;

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        n = scan.nextInt();
        m = scan.nextInt();
        trees = new int[n];
        for(int i=0;i<m;i++){
            int begin = scan.nextInt();
            int end = scan.nextInt();
            int pay = scan.nextInt();
            for(int j=begin-1;j<=end-1;j++){
                trees[j]+=pay;
            }
        }
        scan.close();
        for(int i=0;i<n;i++){
            res+=trees[i];
        }
        System.out.println(res);
    }
}

前缀和

public class 大学里的树木要维护 {
    static int n;//树总数
    static int m;//区间数
    static int[] trees;

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        n = scan.nextInt();
        m = scan.nextInt();
        trees = new int[n];
        for(int i=0;i<n;i++){
            trees[i] = scan.nextInt();
        }
        for(int i=0;i<m;i++){
            int begin = scan.nextInt();
            if(begin==0)    begin++;
            int end = scan.nextInt();
            int ans = 0;
            for(int j=begin-1;j<=end-1;j++){
                ans+=trees[j];
            }
            System.out.println(ans);
        }
        scan.close();
    }
}