[SWEA] 12712 - 파리퇴치 3 (자바/Java)

https://swexpertacademy.com/main/code/userProblem/userProblemDetail.do?contestProbId=AXuARWAqDkQDFARa

 

이차원 배열의 사방탐색과 팔방탐색 유형이다.

먼저 +형태를 훑고, 그 다음 x형태를 훑으며 가장 큰 sum의 값을 업데이트 해주었다.

+형태의 dr, dc 배열과 x형태의 dr, dc배열을 각각 정의하여 루프를 돌아서 sum을 구하였다.

package SWEA0729;

import java.util.Scanner;

public class SWEA_12712_파리퇴치3 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int T = sc.nextInt();
		for (int t = 0; t < T; t++) {
			int N = sc.nextInt();
			int M = sc.nextInt();
			int sum = 0;
			int[][] arr = new int[N][N];
			for (int i = 0; i < N; i++) {
				for (int j = 0; j < N; j++) {
					arr[i][j] = sc.nextInt();
				}
			}

			// + 형태
			int[] dr = { -1, 1, 0, 0 };
			int[] dc = { 0, 0, 1, -1 };
			for (int r = 0; r < N; r++) {
				for (int c = 0; c < N; c++) {
					int tmp = arr[r][c];
					for (int m = 1; m < M; m++) {
						for (int d = 0; d < 4; d++) {
							int nr = r + dr[d] * m;
							int nc = c + dc[d] * m;
							if (nr >= 0 && nr < N && nc >= 0 && nc < N) {
								tmp += arr[nr][nc];
							}
						}

					}
					if (tmp > sum) {
						sum = tmp;
					}
				}
			}

			// x 형태
			int[] drx = { -1, 1, -1, 1 };
			int[] dcx = { -1, 1, 1, -1 };
			for (int r = 0; r < N; r++) {
				for (int c = 0; c < N; c++) {
					int tmp = arr[r][c];
					for (int m = 1; m < M; m++) {
						for (int d = 0; d < 4; d++) {
							int nr = r + drx[d] * m;
							int nc = c + dcx[d] * m;
							if (nr >= 0 && nr < N && nc >= 0 && nc < N) {
								tmp += arr[nr][nc];
							}
						}

					}
					if (tmp > sum) {
						sum = tmp;
					}
				}
			}

			System.out.printf("#%d %d\n", t + 1, sum);

		}
	}

}