[11659] 구간 합 구하기 4 (JAVA)

# 문제 설명

수 N개가 주어졌을 때, i번째 수부터 j번째 수까지 합을 구하는 프로그램을 작성하시오.

---

입력

첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이다. 셋째 줄부터 M개의 줄에는 합을 구해야 하는 구간 i와 j가 주어진다.

---

출력

총 M개의 줄에 입력으로 주어진 i번째 수부터 j번째 수까지 합을 출력한다.

---

제한

• 1 ≤ N ≤ 100,000
• 1 ≤ M ≤ 100,000
• 1 ≤ i ≤ j ≤ N

 

# 정답 코드

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[] numbers = new int[N];
        int[] sum = new int[N];

        for (int i = 0; i < N; i++) {
            numbers[i] = sc.nextInt();

            if (i == 0) {
                sum[i] = numbers[i];
            }
            else {
                sum[i] = sum[i-1] + numbers[i];
            }
        }

        for (int i = 0; i < M; i++) {
            int start = sc.nextInt();
            int end = sc.nextInt();
            int answer;

            if (start != 1) {
                answer = sum[end - 1] - sum[start - 2];
            } else {
                answer = sum[end - 1];
            }
            System.out.println(answer);
        }

        sc.close();
    }
}

 

수를 저장할 배열 numbers와 합을 저장할 sum을 크기를 N으로 하여 생성한다.

 

수의 개수 N만큼 반복하며 numbers에 값을 저장한다.

sum에는 합을 저장하는데, 이때 i가 0이면 numbers[0]이 저장하고 나머지는 합 공식으로 계산하여 저장한다.

 

다음 합을 구해야 하는 횟수 M만큼 반복하며 각 구간의 합을 출력한다.

주어지는 구간은 0부터 시작하지 않고 1부터 시작하므로 end와 start에 1을 뺀다.

만약 start가 1이면 배열의 제일 처음부터 시작하는 것이므로 sum[end-1]를 출력하고 나머지는 구간 합 공식으로 계산하여 출력한다.