[백준 9996번] 한국이 그리울 땐 서버에 접속하지

문제 링크: 9996번 제목

https://www.acmicpc.net/problem/9996

 

파이썬은 어지간해서는 Exception을 발생시키지 않으니 가끔 엣지 케이스를 잡아내기 힘들다는 단점도 있다. 이 문제의 경우, 아래와 같은 코드를 사용할 경우, 길이 체크 부분의 코드가 없으면 "a*a" 패턴에 대해서 "a"라는 파일 이름에 대해서도 결과값을 DA로 내버린다는 함정이 있었다.

 

def compare_string(a, b):
    if len(a) != len(b):
        return False
    for i in range(len(a)):
        if a[i] != b[i]:
            return False
    return True

n = int(input())
p = input()

star = 0
for i in range(len(p)):
    if p[i]=='*':
        star = i
left = p[:star]
right = p[star+1:]

answer = ""

for _ in range(n):
    case = input()
    
    # 길이 체크
    if len(case) < len(left) + len(right):
        answer += "NE\n"
        continue
    
    check_left = compare_string(left, case[:len(left)])
    if not check_left:
        answer += "NE\n"
        continue
    
    check_right = compare_string(right, case[len(case)-len(right):])
    if not check_right:
        answer += "NE\n"
        continue
    
    answer += "DA\n"

print(answer)