[Python]Baekjoon9996: 한국이 그리울 땐 서버에 접속하지

한국이 그리울 땐 서버에 접속하지

1 초

128 MB

6122

1662

1391

27.926%

문제

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

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

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

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

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

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

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

입력

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

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

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

출력

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

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

풀이1

패턴의 형식이 정확히 어떤 모습인지 언급되지 않았습니다. 하지만 별표는 오직 하나이며 양 끝에 존재하지 않으므로

a*b 나 cd*da 같은 모습일 것입니다.

왼쪽과 오른쪽 값 가져오기

python의 split( arg ) 함수를 이용해 왼쪽과 오른쪽의 값을 가져오고 검사하는 방식으로 처리합니다.

함수

입력 받은 문자열이 패턴에 부합하는지 확인하는 함수를 만듭니다.

1) 왼쪽과 오른쪽의 길이를 더한 값보다 짧다면 패턴에 맞지 않습니다

2) 패턴에서 짤라온 왼쪽과 오른쪽이 각각 입력 받은 문자열의 끝값인지 확인합니다.

3) 끝값과 같다면 True, 아니라면 False 를 반환합니다

def isSame(sentence):
    if len(sentence)<len_left+len_right:
        return False
    elif pattern[0] == sentence[0:len_left] and pattern[1] == sentence[-len_right:]:
        return True
    else:
        return False
    
N = int(input())

pattern = input().split('*')
len_left, len_right = len(pattern[0]), len(pattern[1])

while N > 0:
    N -= 1
    if isSame(input()):
        print('DA')
    else:
        print('NE')

 

풀이2

위와 방식은 비슷하나 startswith()과 endswith() 메소드를 활용하는 방법이 있습니다.

N = int(input())
p = input().split('*')
l = len(p[0])+len(p[1])
while N > 0:
    N -= 1
    s = input()
    if len(s) < l:
        print('NE')
    elif s.startswith(p[0]) and s.endswith(p[1]):
        print('DA')
    else:
        print('NE')