[Python] 백준 9996 한국이 그리울 땐 서버에 접속하지 (Brute Force)

📌 문제

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

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

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

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

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

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

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

입력

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

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

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

출력

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

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

예제 입력 1

3
a*d
abcd
anestonestod
facebook

예제 출력 1

DA
DA
NE

예제 입력 2

6
h*n
huhovdjestvarnomozedocisvastan
honijezakon
atila
je
bio
hun

예제 출력 2

DA
DA
NE
NE
NE
DA

📌 풀이

💬 Code

import sys
input = lambda: sys.stdin.readline().rstrip()

n = int(input())
pattern = input()
star = pattern.index('*')
before, after = pattern[:star], pattern[star+1:]

for _ in range(n):
    name = input()

    if before in name and after in name:
        idx_b = name.index(before)
        idx_a = name.index(after)

        if idx_b == 0 and idx_b+len(before) <= idx_a and name[::-1].index(after[::-1]) == 0:
            print('DA')
            continue

    print('NE')

💡 Solution

만약 패턴으로 h*n이 주어지고, 파일 이름으로 honijezakon이 주어지면

• before: 패턴에서 * 이전의 문자열 (→ h)
• after: 패턴에서 * 이후의 문자열 (→ n)
• idx_b: 파일 이름에서 before의 위치 (→ 0)
• idx_a: 파일 이름에서 after의 위치 (→ 10)

if before in name and after in name:
    idx_b = name.index(before)
    idx_a = name.index(after)

만약 파일 이름에 before과 after이 둘 다 들어간다면 idx_b과 idx_a를 구한다. 들어가지 않는다면 index가 작동하지 않으므로 조건문을 하나 걸어준 것이다.

if idx_b == 0 and idx_b+len(before) <= idx_a and name[::-1].index(after[::-1]) == 0:
    print('DA')
    continue

DA를 출력하기 위해 필요한 다음 조건으로는 세 개가 있다.

👉 idx_b == 0

• before가 파일 이름상에서 첫 번째 위치여야 한다.

👉 name[::-1].index(after[::-1]) == 0

• after가 파일 이름상에서 마지막 위치여야 한다.
• 1
  a*bc
  akkbckkbc
  이 경우처럼 after가 파일 이름상에 여러번 등장하는 경우, 마지막에 등장하는 after을 체크해줘야 한다. index 함수는 첫 등장 위치를 반환해주기 때문에 파일 이름과 after을 뒤집어서 체크! (cbkkcbkka에서 cb의 위치는 0)

👉 idx_b + len(before) <= idx_a

• before의 마지막 글자가 after의 첫번째 글자보다 앞쪽에 위치해야 한다.
• 1
  abcd*cdef
  abcdef
  이 조건이 없으면 이 경우를 DA로 체크하게 된다. before-after간 글자 중복체크를 거르기 위해 필요한 조건이다.