파이썬 알고리즘-21 (탐색)격자판 회문수

21.격자판 회문수

1부터 9까지의 자연수로 채워진 7*7 격자판이 주어지면 격자판 위에서 가로방향 또는
세로방향으로 길이 5자리 회문수가 몇 개 있는지 구하는 프로그램을 작성하세요.
회문수란 121과 같이 앞에서부터 읽으나 뒤에서부터 읽으나 같은 수를 말합니다.

2 4 1 5 3 2 6
3 5 1 8 7 1 7
8 3 2 7 1 3 8
6 1 2 3 2 1 1
1 3 1 3 5 3 2
1 1 2 5 6 5 2
1 2 2 2 2 1 5

빨간색처럼 구부러진 경우(87178)는 회문수로 간주하지 않습니다.

▣ 입력설명
1부터 9까지의 자연수로 채워진 7*7격자판이 주어집니다.

▣ 출력설명
5자리 회문수의 개수를 출력합니다.

▣ 입력예제 1
2 4 1 5 3 2 6
3 5 1 8 7 1 7
8 3 2 7 1 3 8
6 1 2 3 2 1 1
1 3 1 3 5 3 2
1 1 2 5 6 5 2
1 2 2 2 2 1 5

▣ 출력예제 1
3

내 코드

lst=[list(map(int, input().split())) for _ in range(7)]

cnt=0
for i in range(7):
    for j in range(3):
        if lst[i][j]==lst[i][j+4] and lst[i][j+1]==lst[i][j+3]:
            cnt+=1
        if lst[j][i]==lst[j+4][i] and lst[j+1][i]==lst[j+3][i]:
            cnt+=1
print(cnt)

일단 테스트에서는 다 맞았는데 풀이랑은 약간 달라서 걱정된다. 만약 이게 맞다면 풀이보다 조금 효율적이지 않을까

풀이

board=[list(map(int, input().split())) for _ in range(7)]
cnt=0
for i in range(3):
    for j in range(7):
        tmp=board[j][i:i+5]
        if tmp==tmp[::-1]:
            cnt+=1
        for k in range(2):
            if board[i+k][j]!=board[i+5-k-1][j]:
                break;
        else:
            cnt+=1
        
print(cnt)

반성점

• 슬라이싱은 생각해봤지만 역순으로 배열해서 비교한다는 생각은 하지 못했다.

배운 것

• 이 문제와는 관계 없지만, 오늘 본 테스트에서 나온건데 입력이 줄바꿈으로 여러 줄 입력되어있는 것을 리스트로 받아 정수형으로 바꿔주는 법:

import sys
lst=sys.stdin.readlines()
lst=[int(i) for i in lst]

• sys.stdin.readlines()는 일반적인 입력 함수인 input()보다 속도가 빠르다고 한다. 그런데 문자열이 필요한 경우 입력받은 각각의 요소에 개행(\n)이 붙어있으므로 끝에 .rstrip()을 붙여준다.