[SW Expert Academy] 11315. 오목 판정

11315. 오목 판정

---

문제)

 

N X N 크기의 판이 있다. 판의 각 칸에는 돌이 있거나 없을 수 있다. 돌이 가로, 세로, 대각선 중 하나의 방향으로 다섯 개 이상 연속한 부분이 있는지 없는지 판정하는 프로그램을 작성하라.

  

[입력]

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.

각 테스트 케이스의 첫 번째 줄에는 하나의 정수 N(5 ≤ N ≤ 20)이 주어진다.

다음 N개의 줄의 각 줄에는 길이 N인 문자열이 주어진다. 각 문자는 ‘o’또는 ‘.’으로, ‘o’는 돌이 있는 칸을 의미하고, ‘.’는 돌이 없는 칸을 의미한다.

  

[출력]

각 테스트 케이스 마다 돌이 다섯 개 이상 연속한 부분이 있으면 “YES”를, 아니면 “NO”를 출력한다.

 

 

 

 

 

이 문제를 풀 때 주의해야 할 점! [반례]

7
.......
o......
.o.....
..o....
...o...
....o..
.......

위 의 경우처럼 1칸씩 좌우로 밀려있는 대각선도 신경을 잘 써야 한다.

나는 이것 때문에 거의 2시간을 고생한 듯...

 

 

 

풀이)

def check1(li): # 행을 확인하는 함수
    for i in range(n):
        cnt = 0
        for j in range(n):
            if li[i][j] == "o":
                cnt += 1
                if cnt == 5:
                    return 1
            elif li[i][j] == ".":
                cnt = 0
    else:
        return 0
 
def check2(li): # ↘ 대각선.
    for i in range(n):
        for j in range(n):
            cnt = 0
            for k in range(5):
                if i + k < n and j + k < n and li[i + k][j + k] == 'o':
                    cnt += 1
                else:
                    break
            if cnt == 5:
                return 1
    else:
        return 0
 
# print(check2(li))
 
def check3(li): # 우측 위로 가는 대각선
    for i in range(n):
        for j in range(n - 1, -1, -1):
            cnt = 0
            for k in range(5):
                if i + k < n and j - k < n and li[i + k][j - k] == 'o':
                    cnt += 1
                else:
                    break
            if cnt >= 5:
                return 1
    else:
        return 0
 
def check4(li): # 열을 확인
    for i in range(n):
        cnt = 0
        for j in range(n):
            if li[j][i] == "o":
                cnt += 1
                if cnt == 5:
                    return 1
            else:
                cnt = 0
    else:
        return 0
 
T = int(input())
for _ in range(T):
    n = int(input())
    li = [list(input()) for i in range(n)]
 
    answer = check1(li) + check2(li) + check3(li) + check4(li)
 
    if answer > 0:
        print(f"#{_ + 1} YES")
    else:
        print(f"#{_ + 1} NO")

 

 

---

문제 출처 : https://swexpertacademy.com/main/main.do

SW Expert Academy

※ SW Expert 아카데미는 원칙적으로 문제를 무단 복제하는 것을 금지합니다.

학습용으로 문제를 가져왔으나, 문제가 될 시 수정 및 삭제하겠습니다.