문제
풀이
- 입력을 받으면서 검은 돌과 흰 돌의 위치를 black과 white vector<pair<int, int>>에 저장한다.
- 검은 돌의 위치를 기준으로 오른쪽으로 오목인지, 아래로 오목인지, 왼쪽 대각선 아래로 오목인지, 왼쪽 대각선 위로 오목인지를 판단한다.
- 이때 각 위치를 기준으로 오목을 판단하기 때문에 6목 이상이라도 그 중간에서부터 오목을 세기 시작하면 오목으로 판단될 수 있기 때문에 판단하기 전에 이전에 이미 판단한 것인지를 돌의 색을 이용하여 판단한다.
- 판단한 결과를 출력하며, 아무것도 판단하지 못한 경우 0을 출력한다.
코드
#include <iostream>
#include <vector>
using namespace std;
int puzzle[21][21] = { 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) {
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) {
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) {
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) {
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;
}