[Baekjoon] 2615_오목

Silver I

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

---

< 오목 >

 

문제 풀이 

 

가로, 세로, 대각선을 살펴보며 다섯 알이 놓였는지 확인한다. 다섯 알이 놓였다면 가장 왼쪽에 있는 바둑알을 출력한다.

 

 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 _2615_ { // 오목
	static int arr[][];

	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;

		arr = new int[20][20];
		for (int i = 1; i <= 19; i++) {
			st = new StringTokenizer(bf.readLine());
			for (int j = 1; j <= 19; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}

		boolean flag = false;
		for (int i = 1; i <= 19; i++) {
			for (int j = 1; j <= 19; j++) {
				if (arr[i][j] != 0) {
					int temp[]=search(i, j, arr[i][j]);
					if (temp != null) {
						flag = true;
						bw.write(arr[i][j] + "\n");
						bw.write(temp[0] + " " + temp[1]);
						break;
					}
				}
			}
			if (flag) {
				break;
			}
		}
		if (!flag) {
			bw.write("0");
		}
		bw.flush();
	}

	private static int[] search(int i, int j, int num) {
		int resultX = 20, resultY = 20;

		// 가로
		int x = i, y = j, cnt = 0;
		while (y < 20) {
			if (arr[x][y] == num) {
				resultY = Math.min(resultY, y);
				cnt += 1;
				y += 1;
			} else {
				break;
			}
		}
		y = j - 1;
		while (y >= 0) {
			if (arr[x][y] == num) {
				resultY = Math.min(resultY, y);
				cnt += 1;
				y -= 1;
			} else {
				break;
			}
		}
		if (cnt == 5) {
			return new int[] { x, resultY };
		}

		// 세로
		resultY = 20;
		y = j;
		cnt = 0;
		while (x < 20) {
			if (arr[x][y] == num) {
				resultX = Math.min(resultX, x);
				cnt += 1;
				x += 1;
			} else {
				break;
			}
		}
		x = i - 1;
		while (x >= 0) {
			if (arr[x][y] == num) {
				resultX = Math.min(resultX, x);
				cnt += 1;
				x -= 1;
			} else {
				break;
			}
		}
		if (cnt == 5) {
			return new int[] { resultX, y };
		}

		// 대각선
		resultX = 20;
		x = i;
		cnt = 0;
		while (x < 20 && y < 20) {
			if (arr[x][y] == num) {
				if (resultY > y) {
					resultX = x;
					resultY = y;
				}
				cnt += 1;
				x += 1;
				y += 1;
			} else {
				break;
			}
		}
		x = i - 1;
		y = j - 1;
		while (x >= 0 && y >= 0) {
			if (arr[x][y] == num) {
				if (resultY > y) {
					resultX = x;
					resultY = y;
				}
				cnt += 1;
				x -= 1;
				y -= 1;
			} else {
				break;
			}
		}
		if (cnt == 5) {
			return new int[] { resultX, resultY };
		}

		resultX = 20;
		resultY = 20;
		x = i;
		y = j;
		cnt = 0;
		while (x < 20 && y >= 0) {
			if (arr[x][y] == num) {
				if (resultY > y) {
					resultX = x;
					resultY = y;
				}
				cnt += 1;
				x += 1;
				y -= 1;
			} else {
				break;
			}
		}
		x = i - 1;
		y = j + 1;
		while (x >= 0 && y < 20) {
			if (arr[x][y] == num) {
				if (resultY > y) {
					resultX = x;
					resultY = y;
				}
				cnt += 1;
				x -= 1;
				y += 1;
			} else {
				break;
			}
		}
		if (cnt == 5) {
			return new int[] { resultX, resultY };
		}
		return null;
	}
}

변수)
arr : 바둑판 정보
flag : 승부 결정 여부
temp : 가장 왼쪽에 있는 바둑알 위치

 

바둑판 정보를 입력받는다. 바둑판을 탐색하면서 흰색이나 검은색 바둑알이 있다면 search 함수를 통해 가로, 세로, 대각선을 탐색한다. 만약 다섯 알이 놓였다면 그 바둑알의 색을 출력하고 가장 왼쪽에 있는 바둑알의 위치를 출력 후 종료한다. 바둑판을 탐색 후에도 승부가 결정되지 않았다면 0을 출력한다.

 

search (int i, int j, int num)

1) 가로를 탐색하기 위해 현재 위치에서 오른쪽, 왼쪽을 살펴본다. 같은 색의 바둑알이 놓였다면 계속해서 확인하고 그렇지 않다면 종료한다. 만약 다섯 알이 놓였다면 가장 왼쪽에 있는 바둑알의 위치를 반환한다.

 

2) 세로를 탐색하기 위해 현재 위치에서 아래, 위를 살펴본다. 같은 색의 바둑알이 놓였다면 계속해서 확인하고 그렇지 않다면 종료한다. 만약 다섯 알이 놓였다면 가장 왼쪽에 있는 바둑알의 위치를 반환한다.

 

3) 대각선을 탐색하기 위해 현재 위치에서 오른쪽 아래, 왼쪽 위를 살펴본다. 같은 색의 바둑알이 놓였다면 계속해서 확인하고 그렇지 않다면 종료한다. 만약 다섯 알이 놓였다면 가장 왼쪽에 있는 바둑알의 위치를 반환한다.

 

4) 대각선을 탐색하기 위해 현재 위치에서 왼쪽 아래, 오른쪽 위를 살펴본다. 같은 색의 바둑알이 놓였다면 계속해서 확인하고 그렇지 않다면 종료한다. 만약 다섯 알이 놓였다면 가장 왼쪽에 있는 바둑알의 위치를 반환한다.

---

---