본문 바로가기
코딩테스트 준비/백준

[백준] 11441번 - 합 구하기 [Java]

by mwzz6 2025. 2. 2.

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

 

[백준] 11441번 - 합 구하기 [Java]
[백준] 11441번 - 합 구하기 [Java]


1.  아이디어

 

구간의 합은 누적합 배열로 간단하게 구할 수 있다.


2. 문제풀이

 

구간의 양 끝을 포함해야해서 A-1로 설정하는 것만 주의해서 구현했다.


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;

        int N = Integer.parseInt(br.readLine());

        int[] arr = new int[1 + N];
        st = new StringTokenizer(br.readLine());
        for (int i = 1; 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];
        }

        int M = Integer.parseInt(br.readLine());

        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. 후기