[Baekjoon] 21236_Comfortable Cows

문제(출처: https://www.acmicpc.net/problem/21236)

---

< Comfortable Cows >

 

문제 풀이 

 

좌표를 입력받을 때마다 상하좌우를 살펴보며 소가 있다면 둘 다 값을 1 더해준다. 이때, 탐색 결과에 따라 인접한 네 방향 중 3마리에 둘러싸인 소의 수를 업데이트한다.

 

my solution (Java)

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

public class _21236_ { // Comfortable Cows

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

		int n = Integer.parseInt(bf.readLine());

		int arr[][] = new int[1001][1001];

		for (int i = 0; i < 1001; i++) {
			for (int j = 0; j < 1001; j++) {
				arr[i][j] = -1;
			}
		}

		int dx[] = { -1, 1, 0, 0 };
		int dy[] = { 0, 0, -1, 1 };

		int result = 0;

		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(bf.readLine());

			int x = Integer.parseInt(st.nextToken());
			int y = Integer.parseInt(st.nextToken());

			arr[x][y] = 0;

			for (int j = 0; j < 4; j++) {
				int x1 = x + dx[j];
				int y1 = y + dy[j];

				if (x1 >= 0 && x1 < 1001 && y1 >= 0 && y1 < 1001) {
					if (arr[x1][y1] > -1) {
						arr[x][y] += 1;

						if (arr[x1][y1] == 3) {
							result -= 1;
						}
						arr[x1][y1] += 1;

						if (arr[x1][y1] == 3) {
							result += 1;
						}
					}
				}
			}

			if (arr[x][y] == 3) {
				result += 1;
			}

			bw.write(result + "\n");
		}
		bw.flush();
	}
}

변수)
n : 좌표 수
arr : 각 위치에 있는 소마다 인접한 방향에 있는 소의 수
dx, dy : 상, 하, 좌, 우
x, y : 좌표
x1, y1 : 상, 하, 좌, 우 위치
result : 인접한 네 방향 중 3마리에 둘러싸인 소의 수

 

좌표 수를 입력받는다. 좌표의 범위가 0 이상 1000 이하 이므로 배열을 선언하여 -1로 초기화한다. 좌표 수만큼 좌표를 입력받아 다음 과정을 거친다.

 

1) 현재 위치 값을 0으로 저장

2) 상하좌우를 탐색하며 범위 안이고 소가 있다면 현재 위치 +1, 탐색 위치에 있는 소도 인접한 소가 늘어나는 것이므로 값을 +1 한다. +1 하기 전 값이 3이라면 result-1, +1 한 후에 값이 3이라면 result +1을 한다.

3) 현재 위치 값이 3이라면 result+1

4) result 출력

---

---