문제:
선영이는 이번 학기에 오스트레일리아로 교환 학생을 가게 되었다.
호주에 도착하고 처음 며칠은 한국 생각을 잊으면서 즐겁게 지냈다. 몇 주가 지나니 한국이 그리워지기 시작했다.
선영이는 한국에 두고온 서버에 접속해서 디렉토리 안에 들어있는 파일 이름을 보면서 그리움을 잊기로 했다. 매일 밤, 파일 이름을 보면서 파일 하나하나에 얽힌 사연을 기억하면서 한국을 생각하고 있었다.
어느 날이었다. 한국에 있는 서버가 망가졌고, 그 결과 특정 패턴과 일치하는 파일 이름을 적절히 출력하지 못하는 버그가 생겼다.
패턴은 알파벳 소문자 여러 개와 별표(*) 하나로 이루어진 문자열이다.
파일 이름이 패턴에 일치하려면, 패턴에 있는 별표를 알파벳 소문자로 이루어진 임의의 문자열로 변환해 파일 이름과 같게 만들 수 있어야 한다. 별표는 빈 문자열로 바꿀 수도 있다. 예를 들어, "abcd", "ad", "anestonestod"는 모두 패턴 "a*d"와 일치한다. 하지만, "bcd"는 일치하지 않는다.
패턴과 파일 이름이 모두 주어졌을 때, 각각의 파일 이름이 패턴과 일치하는지 아닌지를 구하는 프로그램을 작성하시오.
입력:
첫째 줄에 파일의 개수 N이 주어진다. (1 ≤ N ≤ 100)
둘째 줄에는 패턴이 주어진다. 패턴은 알파벳 소문자와 별표(아스키값 42) 한 개로 이루어져 있다. 문자열의 길이는 100을 넘지 않으며, 별표는 문자열의 시작과 끝에 있지 않다.
다음 N개 줄에는 파일 이름이 주어진다. 파일 이름은 알파벳 소문자로만 이루어져 있고, 길이는 100을 넘지 않는다.
출력:
총 N개의 줄에 걸쳐서, 입력으로 주어진 i번째 파일 이름이 패턴과 일치하면 "DA", 일치하지 않으면 "NE"를 출력한다.
참고로, "DA"는 크로아티어어로 "YES"를, "NE"는 "NO"를 의미한다.
풀이방법:
패턴과 일치하는 문자열을 찾기 위해 사용할 수 있는 라이브러리로 정규표현식이 있다. 따라서. 주어진 패턴을 정규표현식 형태로 변경하여 출력을 하도록 한다.
문제에서 주어진 *와 같은 역할을 하는 것은 정규표현식에서 . 에 해당한다. 따라서 replace를 사용하여 이를 변경해주도록 한다. 그리고 패턴이 변수에 할당된 상태로 사용해야 하기 때문에 formatting을 사용하여 string 형태로 사용할 수 있도록 한다. 이 패턴과 추가적으로 ^(맨 앞임을 의미), $(맨 끝임을 의미)를 사용하여 패턴을 마무리하도록 한다. 들어 오는 입력대로 패턴과 일치하는지 확인하여 알맞은 출력을 하도록 한다.
1 2 3 4 5 6 7 8 9 10 11 12 13 | import re n = int(input()) pattern = input() pattern = pattern.replace("*",".*")+"+" for _ in range(n): string = input() p = re.compile(rf"^{pattern}$") m = p.match(string) if m: print("DA") else: print("NE") | cs |
문제링크:
https://www.acmicpc.net/problem/9996
'Algorithm > Python' 카테고리의 다른 글
[BOJ]1940. 주몽 (0) | 2022.06.21 |
---|---|
[BOJ]1213. 팰린드롬 만들기 (0) | 2022.06.16 |
[BOJ]2437. 저울 (0) | 2022.06.02 |
[BOJ]8980. 택배 (0) | 2022.05.31 |
[BOJ]10834. 벨트 (0) | 2022.05.26 |