백준 2615번 오목

31249 단어 ps백준cppcpp

문제

풀이

  1. 입력을 받으면서 검은 돌과 흰 돌의 위치를 black과 white vector<pair<int, int>>에 저장한다.
  2. 검은 돌의 위치를 기준으로 오른쪽으로 오목인지, 아래로 오목인지, 왼쪽 대각선 아래로 오목인지, 왼쪽 대각선 위로 오목인지를 판단한다.
  3. 이때 각 위치를 기준으로 오목을 판단하기 때문에 6목 이상이라도 그 중간에서부터 오목을 세기 시작하면 오목으로 판단될 수 있기 때문에 판단하기 전에 이전에 이미 판단한 것인지를 돌의 색을 이용하여 판단한다.
  4. 판단한 결과를 출력하며, 아무것도 판단하지 못한 경우 0을 출력한다.

코드

//난이도 : 실버2
#include <iostream>
#include <vector>

using namespace std;

int puzzle[21][21] = { 0 }; //왼쪽, 오른쪽, 위, 아래에 0으로 테두리 만듬
vector<pair<int, int>> white;
vector<pair<int, int>> black;

bool right_check(int r, int c, int color) { //오른쪽으로 오목 확인
	if (puzzle[r][c - 1] == color) { //이전에 검사 했는지 확인
		return false;
	}
	int cnt = 0;
	int i = 0;
	while (puzzle[r][c + i] == color) { //같은 색의 돌의 수 세기
		cnt++;
		i++;
	}
	if (cnt == 5) { //5개일 때만 정답
		return true;
	}
	else {
		return false;
	}
}
bool bottom_check(int r, int c, int color) { //아래로 오목 확인
	if (puzzle[r - 1][c] == color) { //이전에 검사 했는지 확인
		return false;
	}
	int cnt = 0;
	int i = 0;
	while (puzzle[r + i][c] == color) { //같은 색의 돌의 수 세기
		cnt++;
		i++;
	}
	if (cnt == 5) { //5개일 때만 정답
		return true;
	}
	else {
		return false;
	}
}
bool diagonal_down_check(int r, int c, int color) { //왼쪽 대각선 아래로 오목 확인
	if (puzzle[r - 1][c - 1] == color) { //이전에 검사 했는지 확인
		return false;
	}
	int cnt = 0;
	int i = 0;
	while (puzzle[r + i][c + i] == color) { //같은 색의 돌의 수 세기
		cnt++;
		i++;
	}
	if (cnt == 5) { //5개일 때만 정답
		return true;
	}
	else {
		return false;
	}
}

bool diagonal_up_check(int r, int c, int color) { //왼쪽 대각선 위로 오목 확인
	if (puzzle[r + 1][c - 1] == color) {
		return false;
	}
	int cnt = 0;
	int i = 0;
	while (puzzle[r - i][c + i] == color) { //같은 색의 돌의 수 세기
		cnt++;
		i++;
	}
	if (cnt == 5) { //5개일 때만 정답
		return true;
	}
	else {
		return false;
	}
}

vector<pair<int, int>> full_check(int r, int c, int *color) { //전체 경우 체크 함수
	vector<pair<int, int>> result;
	
	if (right_check(r, c, *color)) { //오른쪽 체크
		result.push_back(make_pair(r, c));
		return result;
	}
	else if (bottom_check(r, c, *color)) { //아래 체크
		result.push_back(make_pair(r, c));
		return result;
	}
	else if (diagonal_down_check(r, c, *color)) { //왼쪽 아래 대각선 체크
		result.push_back(make_pair(r, c));
		return result;
	}
	else if (diagonal_up_check(r, c, *color)) { //왼쪽 위 대각선 체크
		result.push_back(make_pair(r, c));
		return result;
	}

	result.push_back(make_pair(-1, -1)); //아무것도 못찾은 경우 
	return result;
}



int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	 //입력
	for (int r = 1; r <= 19; r++) {
		for (int c = 1; c <= 19; c++) {
			cin >> puzzle[r][c];
			if (puzzle[r][c] == 1) { //검은 돌 위치 저장
				black.push_back(make_pair(r, c));
			}
			else if (puzzle[r][c] == 2) { //흰 돌 위치 저장
				white.push_back(make_pair(r, c));
			}
		}
	}
	
	int color = 1; //검은 돌
	vector<pair<int, int>> result;
	for (int i = 0; i < black.size(); i++) { //검은 돌에서 오목 확인
		result = full_check(black[i].first, black[i].second, &color);//오목 확인
		if (result[0].first != -1 && result[0].second != -1) { //찾은 경우
			cout << color << '\n';
			cout << result[0].first << " " << result[0].second << '\n';
			return 0; // 종료
		}
	}

	color = 2; //흰 돌
	for (int i = 0; i < white.size(); i++) { //흰 돌에서 오목 확인
		result = full_check(white[i].first, white[i].second, &color);//오목 확인
		if (result[0].first != -1 && result[0].second != -1) { //찾은 경우
			cout << color << '\n';
			cout << result[0].first << " " << result[0].second << '\n';
			return 0; //종료
		}
	}
	
	//승부가 안난 경우
	cout << "0" << '\n';
	

	return 0;
}

좋은 웹페이지 즐겨찾기