算法训练营——day5数组简单前缀和

卡玛网（区间和）-简单题

题目描述

给定一个整数数组 Array，请计算该数组在每个指定区间内元素的总和。

输入描述

第一行输入为整数数组 Array 的长度 n，接下来 n 行，每行一个整数，表示数组的元素。随后的输入为需要计算总和的区间下标：a，b （b > = a），直至文件结束。

输出描述

输出每个指定区间内元素的总和。

输入示例

5
1
2
3
4
5
0 1
1 3

输出示例

3
9

提示信息

数据范围：
0 < n <= 100000

前缀和Java与C嘎嘎

#include <iostream>
#include <vector>
using namespace std;
int main() {
    int n, a, b;
    cin >> n;
    vector<int> vec(n);
    vector<int> flag(n);
    int presum = 0;
    for (int i = 0; i < n; i++) {
        cin >> vec[i];
        subsum += vec[i];
        flag[i] = subsum;
    }

    while (cin >> a >> b) {
        int sum;
        if (a == 0) sum = flag[b];
        else sum = flag[b] - flag[a - 1];
        cout << sum << endl;
    }
}

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        int n = scanner.nextInt();
        int[] ret = new int[n];
        int[] flag = new int[n];

        int presum = 0;
        for (int i = 0; i < n; i++) {
            ret[i] = scanner.nextInt();
            presum += ret[i];
            flag[i] = presum;
        }

        while (scanner.hasNextInt()) {
            int a = scanner.nextInt();
            int b = scanner.nextInt();

            int sum;
            if (a == 0) {
                sum = flag[b];
            } else {
                sum = flag[b] - flag[a - 1];
            }
            System.out.println(sum);
        }

        scanner.close();
    }
}