[파이썬 알고리즘] 격자판 회문수

격자판 회문수

📄 문제

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

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

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

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

▣ 입력예제
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

▣ 출력예제
3

🧞 풀이

풀이

'''
가로로 검사해야 하는 수: 3 * 7 = 21
세로로 검사해야 하는 수: 3 * 7 = 21
42개 검사할 수 있는 함수를 만들면 됨

[0:5]
'''

cnt = 0                                                         # 갯수 세기 위한 cnt 할당

a = [list(map(int, input().split())) for _ in range(7)]         # 배열 값들 입력하여 할당

tmp = []                                                        # 행 슬라이싱해서 담을 빈 리스트 할당
for i in range(7):                                              # 7번 반복
    for j in range(3):                                          # 3번 반복 (이중반복)
        tmp = a[i][0+j:5+j]                                     # 5개의 요소만 슬라이싱

        if tmp[0] == tmp[4] and tmp[1] == tmp[3]:               # 회문일 때,
            cnt += 1                                            # cnt 1 증가

        if a[0+j][i] == a[4+j][i] and a[1+j][i] == a[3+j][i]:   # 세로로 회문일 때,
            cnt += 1                                            # cnt 1 증가

print(cnt)                                                      # 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)