본문 바로가기
Computer/Algorithm_Java

[BaekJoon] 11659번 구간 합 구하기4 문제 풀이 (Success)

by HJ0216 2023. 9. 10.
 

11659번: 구간 합 구하기 4

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

www.acmicpc.net

 

Language: Java

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
        StringTokenizer st = new StringTokenizer(br.readLine());
 
        int N = Integer.valueOf(st.nextToken());
        int M = Integer.valueOf(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());
        }
 
        StringBuilder sb = new StringBuilder();
 
        while (M-- > 0) {
            st = new StringTokenizer(br.readLine());
            int start = Integer.valueOf(st.nextToken()) - 1;
            int end = Integer.valueOf(st.nextToken()) - 1;
            int sum = 0;
            for (int j = start; j <= end; j++) {
                sum += arr[j];
            }
            sb.append(sum).append("\n");
        }
 
        bw.write(sb.toString());
 
        bw.flush();
        bw.close();
 
    }
}
 
 

🚨 시간 초과

- while (M-- > 0) { for(...){ ... } }

    - 2중 반벅문으로 인한 시간 초과

 

😮 찾아본 풀이

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
        StringTokenizer st = new StringTokenizer(br.readLine());
 
        int N = Integer.valueOf(st.nextToken());
        int M = Integer.valueOf(st.nextToken());
 
        int[] arr = new int[N+1];
 
        st = new StringTokenizer(br.readLine());
        for (int i = 1; i <= N; i++) {
            arr[i] = arr[i - 1+ Integer.parseInt(st.nextToken());
        }
 
        while (M-- > 0) {
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken()) - 1;
            int end = Integer.parseInt(st.nextToken());
            int sum = arr[end] - arr[start];
            bw.write(sum + "\n");
        }
 
        bw.flush();
        bw.close();
 
    }
}
 
 

1. arr[i] = arr[i - 1] + Integer.parseInt(st.nextToken());

    - 배열에 누적 합을 저장

2. int sum = arr[end] - arr[start];

    - 구간 합을 누적 합에서 계산

 

 

 

🔗 소스 코드
GitHub

 

📚 참고 자료

 

[JAVA/백준] 11659번: 구간 합 구하기 4

PS & 개발 기록

iamheesoo.github.io