SWEA: 1215 회문1

문제

SW Expert Academy

 

문제 풀이

 행열의 가로 세로의 문자열을 입력된 길이만큼 split 하여 회문이 되는지 확인하고 가능한 경우를 반환하는 문제이다. 회문이 되기 위해서는 `string == string[::-1]`을 만족하면 된다. 세로 탐색을 쉽게 하기 위해 앞서 다룬 Sum 문제와 같이 행열을 전치하여 문제를 풀었다.

 

코드

for test_case in range(1, 11):
    answer = 0
    length = int(input())
    array = [list(input().strip()) for _ in range(8)]
    trans_array = list(map(list, zip(*array)))

    for i in range(8):
        for j in range(9 - length):
            garo = array[i][j:j + length]
            sero = trans_array[i][j:j + length]

            if garo == garo[::-1]:
                answer += 1

            if sero == sero[::-1]:
                answer += 1

    print('#' + str(test_case), answer)