4861. [파이썬 S/W 문제해결 기본] 3일차 - 회문

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&contestProbId=AWTQQXcKQHkDFAVT&categoryId=AWTQQXcKQHkDFAVT&categoryType=CODE&problemTitle=%ED%9A%8C%EB%AC%B8&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=2&pageSize=10&pageIndex=1

SW Expert Academy

뭔가 어렵다고 느껴졌는데 생각보다 단순한 알고리즘이였다.

T=int(input())
for tc in range(1,T+1):
    #1.입력받기 #2.자료형 정의
    N,M=map(int,input().split()) #N:matrix NxN, M:회문길이
    arr=[]
    for _ in range(N):
        arr.append(list(input()))
    ans=[]

    #3.동작함수
    #가로체크
    for i in range(N):
        for j in range(N-M+1):
            rowstr=arr[i][j:j+M] #가로
            if rowstr==rowstr[::-1]:
                ans=rowstr
                break
        if ans:
            break
    #세로체크
    if not ans:
        for i in range(N-M+1):
            for j in range(N):
                colstr=[] #세로
                for m in range(M):
                    colstr.append(arr[i+m][j])
                if colstr==colstr[::-1]:
                    ans=colstr
                    break
            if ans:
                break
    #4.출력
    print(f"#{tc} {''.join(map(str,ans))}")

Python에서 sub == sub[::-1] 은 문자열을 거꾸로 출력해주는 표현식이였다는 것을 몰랐어서 좀 더 풀어서 코딩했었다.

 

<문자열 거꾸로 출력하기>

s = 'ABCDE'
print(s[::-1])  #'EDCBA'

 

아래 코드는 처음 풀었을 때의 코드이다. 

T=int(input())
for tc in range(1,T+1):
    #1.입력받기 #2.자료형 정의
    N,M=map(int,input().split()) #N:matrix NxN, M:회문길이
    arr=[]
    for _ in range(N):
        arr.append(list(input()))
    ans=[]

    #3.동작함수
    #가로체크
    for i in range(N):
        for j in range(N-M+1):
            rowstr=arr[i][j:j+M] #가로
            check_row=True #회문 체크
            for m in range(M):
                if rowstr[m]!=rowstr[-1-m]:
                    check_row=False
            if check_row==True:
                ans=rowstr
    #세로체크
    for i in range(N-M+1):
        for j in range(N):
            colstr=[] #세로
            for m in range(M):
                colstr.append(arr[i+m][j])
            check_col=True #회문 체크
            for m in range(M):
                if colstr[m]!=colstr[-1-m]:
                    check_col = False
            if check_col==True:
                ans=colstr
    #4.출력
    print(f"#{tc} {''.join(map(str,ans))}")