백준 9996번 파이썬 코드 답안

문제

선영이는 이번 학기에 오스트레일리아로 교환 학생을 가게 되었다. 

호주에 도착하고 처음 며칠은 한국 생각을 잊으면서 즐겁게 지냈다. 몇 주가 지나니 한국이 그리워지기 시작했다. 

선영이는 한국에 두고온 서버에 접속해서 디렉토리 안에 들어있는 파일 이름을 보면서 그리움을 잊기로 했다. 매일 밤, 파일 이름을 보면서 파일 하나하나에 얽힌 사연을 기억하면서 한국을 생각하고 있었다.

어느 날이었다. 한국에 있는 서버가 망가졌고, 그 결과 특정 패턴과 일치하는 파일 이름을 적절히 출력하지 못하는 버그가 생겼다.

패턴은 알파벳 소문자 여러 개와 별표(*) 하나로 이루어진 문자열이다.

파일 이름이 패턴에 일치하려면, 패턴에 있는 별표를 알파벳 소문자로 이루어진 임의의 문자열로 변환해 파일 이름과 같게 만들 수 있어야 한다. 별표는 빈 문자열로 바꿀 수도 있다. 예를 들어, "abcd", "ad", "anestonestod"는 모두 패턴 "a*d"와 일치한다. 하지만, "bcd"는 일치하지 않는다.

패턴과 파일 이름이 모두 주어졌을 때, 각각의 파일 이름이 패턴과 일치하는지 아닌지를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 파일의 개수 N이 주어진다. (1 ≤ N ≤ 100)

둘째 줄에는 패턴이 주어진다. 패턴은 알파벳 소문자와 별표(아스키값 42) 한 개로 이루어져 있다. 문자열의 길이는 100을 넘지 않으며, 별표는 문자열의 시작과 끝에 있지 않다.

다음 N개 줄에는 파일 이름이 주어진다. 파일 이름은 알파벳 소문자로만 이루어져 있고, 길이는 100을 넘지 않는다.

출력

총 N개의 줄에 걸쳐서, 입력으로 주어진 i번째 파일 이름이 패턴과 일치하면 "DA", 일치하지 않으면 "NE"를 출력한다.

참고로, "DA"는 크로아티어어로 "YES"를, "NE"는 "NO"를 의미한다.

문제풀이

먼저 문제를 정확하게 이해를 해야한다. 'h*n'이라는 패턴이 주어진다면, 반드시 문자는 h로 시작을 해야하고, 끝이 n으로 나야한다. 예를 들어 h*n이라는 패턴이 주어졌을 때, haaaan은 'DA'지만 haanaaa는 'NE'다. 그 이유는 n으로 끝나지 않기 때문이다. 이걸 정확하게 이해하지 않으면, 아무리 풀어도 오답이 나는 경우가 발생한다. 

문제를 정확하게 이해하면, 문제 풀이는 쉽다. 먼저 경우의 수를 나눈다. 항상 *는 문자열에 가운데에 온다고 했으나, * 앞뒤로 오는 문자열의 갯수는 정해져 있지 않다고 한다. 즉, 별표 앞,뒤로 오는 문자열은 각각 1개 이상일 수 있다. 따라서, 해당 경우를 충분히 고려해주어야 한다. 나 같은 경우에는 Flag를 활용해서 진행을 했지만, 사실은 split()메서드를 활용해 충분히 해결 가능하다.

>> temp = str(sys.stdin.readline().rstrip()
>> c,d = temp.split('*')

별표 앞,뒤로 오는 문자열을 구분해두었으면 그 다음부터 나는 다음과 같이 판별을 했다. 먼저 앞의 문자열이 가장 앞에 있는지를 내장 메소드인 .find()를 사용해서 동일 패턴이 시작되는 첫번째 index를 찾고 그 값을 저장했다. 그리고 문자열을 하나 카피했는데, 동일 패턴이 시작되는 첫번째 index와 시작 패턴의 길이만큼 뒷쪽부터 문자열의 끝까지를 복사해서 문자열을 저장했다. 그리고 문자열을 반전해서 다시 저장했다. 반전한 문자열에서 마지막 문자열 패턴이 시작되는 위치가 '0'인 것을 확인하다면, 해당 문자열로 시작하고 끝나는 것으로 판단했다.  코드는 아래와 같다.

import sys
t = int(sys.stdin.readline().rstrip())
temp = str(sys.stdin.readline().rstrip())
flag = 'F'
pre_s =''
post_s =''
for i in temp :
    if i == '*' :
        flag = 'T'
    elif flag == 'F' :
        pre_s += i
    elif flag == 'T' :
        post_s += i

########### 위의 for문은 아래 코드로 단순화 가능. 
#pre_s , post_s = temp.split('*')


post_s = post_s[::-1]

for i in range(t) :
    k = - 1
    j = - 1
    temp = str(sys.stdin.readline().rstrip())
    k = temp.find(pre_s)
    h = temp[k+len(pre_s):][::-1]
    j = h.find(post_s)
    if k == 0 and j == 0 :
        print('DA')
    else :
        print('NE')