三种语言实现前缀和(C++/Python/Java)

题目

输入一个长度为 n 的整数序列。

接下来再输入 m 个询问,每个询问输入一对 l,r

对于每个询问,输出原序列中从第 l个数到第 r 个数的和。

输入格式

第一行包含两个整数 n 和 m。

第二行包含 n 个整数,表示整数数列。

接下来 m 行,每行包含两个整数 l 和 r,表示一个询问的区间范围。

输出格式

共 m 行,每行输出一个询问的结果。

数据范围

1≤l≤r≤n
1≤n,m≤100000
−1000≤数列中元素的值≤1000

输入样例:

5 3
2 1 3 6 4
1 2
1 3
2 4

输出样例:

3
6
10

C++

#include <bits/stdc++.h>

using namespace std;

const int N = 100010;
int a[N], s[N];

int main()
{
    int n,m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        cin >> a[i];
        s[i] = s[i-1] + a[i];
    }
        
    while(m--)
    {
        int l,r;
        cin >> l >> r;
        cout << s[r] - s[l-1] << endl;
    }
}

Python

n,m = map(int, input().split())
a = [int(x) for x in input().split()]
a = [0] + a
s = [0]
for i in range(1, n+1):
    s.append(s[i-1] + a[i])
while m > 0:
    m -= 1
    l, r = map(int, input().split())
    print(s[r] - s[l-1])

Java

import java.util.*;
 
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int[] a = new int[n+1];
        int[] s = new int[n+1];
        for(int i = 1; i <= n; i++)
        {
            a[i] = sc.nextInt();
            s[i] = s[i-1] + a[i];
        }
        while(m > 0){
            m -= 1;
            int l = sc.nextInt();
            int r = sc.nextInt();
            System.out.println(s[r] - s[l-1]);
        }
    }
}
posted @ 2024-07-31 10:08  火车驶向云外0218  阅读(37)  评论(0)    收藏  举报