SWEA - 회문[Python]

😅 문제 

https://swexpertacademy.com/main/

🤔 문제 상황 

- 회문이 가로, 세로 두가지 경우가 발생할 수 있다. 

 

🧐 해결 전략 

- 회문을 탐색하는 함수를 만들어 판별한다. 

🎰 코드 

def palin(s):
    rev = s[::-1] # s를 거꾸로 출력
    if s[:len(s)//2] == rev[:len(s)//2]:
        return True
    return False
 
t = int(input()) # testcase 개수
result = [] # 결과 리스트
for _ in range(t):
    n , m = map(int,input().split()) # n x n 행렬의 길이, m = 회문의 길이
    nlist = []
    isCheck = False
    isCheck2 = False
    for i in range(n): # n 행을 입력받음
        nlist.append(input())
    # 회문이 가로로 존재
    for i in range(n):
        temp = nlist[i]    
        for j in range(n-m+1):
            if palin(temp[j:j+m]):
                result.append(temp[j:j+m])
                isCheck = True
                isCheck2 = True
                break    
        if isCheck :
            break
    if isCheck2 :
        continue
    # 회문이 세로로 존재
    for i in range(n):
        isCheck3 = False
        temp = []
        for j in range(n):
            temp.append(nlist[j][i])
        for j in range(n-m+1):
            if palin(temp[j:j+m]):
                result.append(''.join(temp[j:j+m]))
                isCheck3 = True
                break    
        if isCheck3 :
            break
for i in range(t):
    print("#{} {}".format(i+1, result[i]))    
 
 

 

🧙‍♂️ 해설

- 우선 가로로 모든 자료를 넣고, 우선 가로를 탐색하며 회문이 나오면 탐색을 종료한다. 가로에 회문이 존재하지 않을 경우 세로로 새로운 배열을 생성하며 회문을 탐색한다. 

 

📈 새로 학습한 것 & 실수 

- 처음에 회문 함수를 만들 때, 변수를 (str, l)로 하여 함수 안에서 모든 것을 해결하려고 하다보니 오히려 복잡하였다. 함수는 깔끔하게 만들고 반복을 통해 찾는게 나을 수도 있다는 것을 알았다.