[SWEA/Java] 2001. 파리 퇴치

[SWEA/Java] 2001. 파리 퇴치

SW Expert Academy

---

해설

풀이

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

public class Solution {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        // 테스트 케이스 수 입력
        int testCase = Integer.parseInt(br.readLine());

        // 각 테스트 케이스에 대한 처리
        for (int t = 1; t <= testCase; t++) {
            // 배열의 크기와 파리채의 크기 입력
            String input = br.readLine();
            StringTokenizer st = new StringTokenizer(input);
            int n = Integer.parseInt(st.nextToken()); // 배열의 크기
            int m = Integer.parseInt(st.nextToken()); // 파리채의 크기

            // 배열 초기화 및 입력
            int[][] graph = new int[n][n];
            for (int r = 0; r < n; r++) {
                input = br.readLine();
                st = new StringTokenizer(input);
                for (int c = 0; c < n; c++) {
                    graph[r][c] = Integer.parseInt(st.nextToken());
                }
            }

            int max = 0;
            // 파리채를 옮기면서 죽인 파리 수 계산
            for (int r = 0; r < n - m + 1; r++) {
                for (int c = 0; c < n - m + 1; c++) {
                    int sum = 0;
                    // 파리채 영역 내의 파리 수 합산
                    for (int i = r; i < r + m; i++) {
                        for (int j = c; j < c + m; j++) {
                            sum += graph[i][j];
                        }
                    }
                    // 최대값 갱신
                    max = Math.max(max, sum);
                }
            }

            // 결과 출력
            sb.append("#" + t + " " + max + "\n");
        }
        // 최종 출력
        System.out.print(sb);
    }
}