코테/백준

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

imname1am 2023. 4. 3. 13:11
반응형

🔺 문제

 

11659번: 구간 합 구하기 4

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

www.acmicpc.net

 

🔺 코드

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

public class Main {
	public static void main(String[] args) throws IOException {
	    var br = new BufferedReader(new InputStreamReader(System.in));
	    var st = new StringTokenizer(br.readLine(), " ");
	    
	    int n = Integer.parseInt(st.nextToken());
	    int m = Integer.parseInt(st.nextToken());
	    int[] arr = new int[n];     // N개의 수
	    int[] sumArr = new int[n];  // 구간 합 배열
	    
	    // N개의 수
	    st = new StringTokenizer(br.readLine(), " ");
	    for(int i = 0 ; i < n ; i++) {
	        arr[i] = Integer.parseInt(st.nextToken());
	    }
	    
	    // 구간 합 구하기
	    sumArr[0] = arr[0];
	    for(int i = 1 ; i < n ; i++) {
	        sumArr[i] = sumArr[i-1] + arr[i];
	    }
	    
	    // 구간 i,j
	    for(int i = 0 ; i < m ; i++) {
	        st = new StringTokenizer(br.readLine(), " ");
	        
	        int a = Integer.parseInt(st.nextToken()) -1;
	        int b = Integer.parseInt(st.nextToken()) -1;
	        
	        if(a == 0) {
	            System.out.println(sumArr[b]);
	        } else {
	            System.out.println(sumArr[b] - sumArr[a - 1]);
	        }
	    }
	}
}
✅ 해결 아이디어
- 구간 합 / 누적 합
- 구간 시작점이 1일 경우 출력 결과 따로 처리

맞긴 맞았는데 메모리 넘 크고 반복문을 세번이나 돌렸다,,,,ㅋ

효율적인 코드를 찾아서,,,,,~


🔺 다른 풀이들

 

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

PS & 개발 기록

iamheesoo.github.io

 

[구간합, JAVA]백준 - 11659번

문제: https://www.acmicpc.net/problem/11659 11659번: 구간 합 구하기 4 첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이

life-study-1031.tistory.com

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

public class Main {
	public static void main(String[] args) throws IOException {
	    var br = new BufferedReader(new InputStreamReader(System.in));
	    var st = new StringTokenizer(br.readLine(), " ");
	    
	    int n = Integer.parseInt(st.nextToken());   // 표 크기
	    int m = Integer.parseInt(st.nextToken());   // 반복 휫수
	    int[] sum = new int[n+1];
	    
	    st = new StringTokenizer(br.readLine(), " ");
	    for(int i = 1 ; i <= n ; i++) {
	        sum[i] = sum[i-1] + Integer.parseInt(st.nextToken());
	    }
	    
	    for(int i = 0 ; i < m ; i++) {
	        st = new StringTokenizer(br.readLine(), " ");
	        int a = Integer.parseInt(st.nextToken());
	        int b = Integer.parseInt(st.nextToken());
	        
	        System.out.println(sum[b] - sum[a-1]);
	    }
	}
}

누적합 계산은 한 번만 하시고, 필요한 부분 뽑아쓰셨다.

입력과 동시에 누적합을 구해서 저장하시는..!! (누적합 배열 크기는 n + 1로 하고, 반복문에서 i 시작 범위는 1부터 시작하게)

다음엔 꼭 이렇게 해야지....!!!!

 

 


(+ 6/7 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
40
41
42
43
44
import java.util.*;
import java.io.*;
 
public class Main {
    static int N, M;
    static long[] arr, sum;
    static int s, e;
    static long result = 0;
    
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        StringBuilder sb = new StringBuilder();
        
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        
        arr = new long[N + 1];
        sum = new long[N + 1];
        
        // 숫자 배열 arr 입력받기
        st = new StringTokenizer(br.readLine(), " ");
        for(int i = 1 ; i <= N ; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
        
        // 구간 합 배열 sum 채우기
        sum[1= arr[1];
        for(int i = 2 ; i <= N ; i++) {
            sum[i] = sum[i-1+ arr[i];
        }
        
        // 구간 값 입력 받아 결과 출력하기
        for(int i = 0 ; i < M ; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            s = Integer.parseInt(st.nextToken());
            e = Integer.parseInt(st.nextToken());
            
            result = (s == e) ? arr[s] : (sum[e] - sum[s - 1]);
            sb.append(result).append("\n");
        }
        System.out.println(sb);
    }
}
cs

반응형