알고리즘/BOJ - Java
[BOJ/Java] 11660. 구간 합 구하기 5
ImJay
2024. 2. 4. 16:38
반응형
[BOJ/Java] 11660. 구간 합 구하기 5
11660번: 구간 합 구하기 5
첫째 줄에 표의 크기 N과 합을 구해야 하는 횟수 M이 주어진다. (1 ≤ N ≤ 1024, 1 ≤ M ≤ 100,000) 둘째 줄부터 N개의 줄에는 표에 채워져 있는 수가 1행부터 차례대로 주어진다. 다음 M개의 줄에는 네
www.acmicpc.net
풀이
package edu.ssafy.im.BOJ.S1.No11660;
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));
StringBuilder sb = new StringBuilder();
// 입력 받기
String input = br.readLine();
StringTokenizer st = new StringTokenizer(input);
int n = Integer.parseInt(st.nextToken()); // 표의 크기 N
int m = Integer.parseInt(st.nextToken()); // 합을 구해야 하는 횟수 M
// 표 초기화
int[][] arr = new int[n][n];
for (int i = 0; i < arr.length; i++) {
input = br.readLine();
st = new StringTokenizer(input);
for (int j = 0; j < arr.length; j++) {
arr[i][j] = Integer.parseInt(st.nextToken());
}
}
// 누적 합 배열 생성
int[][] sum = new int[n][n + 1];
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
sum[i][j + 1] = sum[i][j] + arr[i][j];
}
}
// 구간 합 구하기
for (int i = 0; i < m; i++) {
input = br.readLine();
st = new StringTokenizer(input);
int x1 = Integer.parseInt(st.nextToken()) - 1; // 구간의 시작 행
int y1 = Integer.parseInt(st.nextToken()) - 1; // 구간의 시작 열
int x2 = Integer.parseInt(st.nextToken()) - 1; // 구간의 끝 행
int y2 = Integer.parseInt(st.nextToken()); // 구간의 끝 열
int ans = 0;
for (int j = x1; j <= x2; j++) {
ans += sum[j][y2] - sum[j][y1];
}
sb.append(ans).append("\n");
}
System.out.print(sb);
}
}
반응형