[백준] 11659번 - 구간 합 구하기 4 [Java]

https://www.acmicpc.net/problem/11659

 

[백준] 11659번 - 구간 합 구하기 4 [Java]

[백준] 11659번 - 구간 합 구하기 4 [Java]

---

1.  아이디어

 

누적합 배열을 활용하면 간단하게 해결할 수 있다.

---

2. 문제풀이

 

주어진 수열에 대해 특정 구간의 합을 반복해서 구하는 문제로 누적합 배열의 두 원소의 차로 구간 합을 간단하게 구할 수 있는 점을 활용해서 구현했다.

---

3. 코드

 

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        int[] arr = new int[N];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        int[] prefixSum = new int[1 + N];
        for (int i = 1; i <= N; i++) {
            prefixSum[i] = prefixSum[i - 1] + arr[i - 1];
        }

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());

            sb.append(prefixSum[b] - prefixSum[a - 1]).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기