SWEA2805 농작물 수확하기

이 문제를 보면 BFS를 이용하거나 2차원 배열의 인덱스를 이용해 마름모에 포함되는지 확인하는 방법이 떠오른다.

BFS가 익숙해지고 보면 단순히 BFS를 이용하는 게 구현하는 것이 쉽지만,

실행시간에서는 인덱스를 이용해 조건을 만들어 주는 것이 훨씬 효율적이다.

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

class Solution {
	static BufferedReader br;
	static BufferedWriter bw;
	static StringTokenizer st;
	static char[][] map;

	public static void main(String args[]) throws Exception {
		br = new BufferedReader(new InputStreamReader(System.in));
		bw = new BufferedWriter(new OutputStreamWriter(System.out));
		int T = Integer.parseInt(br.readLine());
		for (int tc = 1; tc <= T; tc++) {
			int N = Integer.parseInt(br.readLine());
			int count = 0;
			map = new char[N][N];
			for (int i = 0; i < N; i++) {
				map[i] = br.readLine().toCharArray();
			}

			for (int i = 0; i < N; i++) {

				for (int j = 0; j < N; j++) {
					int dx = Math.abs(i - N / 2);
					int dy = Math.abs(j - N / 2);
					if (dx + dy > N / 2) {
						continue;
					} else {
						count += map[i][j] - '0';
					}
				}
			}

			bw.write(String.format("#%d ", tc));
			bw.write(String.format("%d\n", count));
		}
		bw.flush();
		bw.close();
	}
}