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

백준 #11659구간 합 구하기4

11659번: 구간 합 구하기 4

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		int n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());
		
		// 누적합 구하기
		st = new StringTokenizer(br.readLine());
		int[] cumSum = new int[n+1];
		for (int i = 1; i < cumSum.length; i++) {
			cumSum[i] = cumSum[i-1] + Integer.parseInt(st.nextToken());
		}
		
		for (int k = 0; k < m; k++) {
			st = new StringTokenizer(br.readLine());
			int i = Integer.parseInt(st.nextToken());
			int j = Integer.parseInt(st.nextToken());
			System.out.println(cumSum[j] - cumSum[i-1]);
		}
		
		
	}  // end of main
}  // end of class

구간마다 합을 새로 구하면 시간 초과가 발생한다. 따라서 누적합을 이용해야 한다! 1부터 해당 자리까지의 누적합을 cumSum에 구해준다. i ~ j의 합은 j 까지의 합 - (i-1) 까지의 합이 된다.

• 저장 O(n)
• 누적합 조회 O(1)
• 값 수정 O(n)