백준: 14426 접두사 찾기

문제

14426번: 접두사 찾기

 

문제 풀이

 M개의 문자들 중 N개의 문자의 접미사가 하나라도 되는지 확인하는 문제이다. 접미사 이므로 찾고자 하는 패턴의 수와 N개의 문자 중 일치하는지 확인하면 쉽게 해결할 수 있다.

 

코드

split을 활용한 풀이

from sys import stdin


if __name__ == '__main__':
    n, m = map(int, stdin.readline().split())
    strings = [stdin.readline().strip() for _ in range(n)]
    cnt = 0

    for _ in range(m):
        pattern = stdin.readline().strip()
        for s in strings:
            if pattern == s[:len(pattern)]:
                cnt += 1
                break
    print(cnt)

 

startswith을 활용한 풀이

from sys import stdin


if __name__ == '__main__':
    n, m = map(int, stdin.readline().split())
    strings = {stdin.readline().strip() for _ in range(n)}
    cnt = 0

    for _ in range(m):
        pattern = stdin.readline().strip()
        for s in strings:
            if s.startswith(pattern):
                cnt += 1
                break
    print(cnt)

 두 방식은 동일하지만 이번에 문제를 풀면서 `startswith`이라는 메소드를 처음 알게 되었다. 방식은 동일하지만 `split`은 712ms, `startswith`은 932ms의 시간이 소요되었다. 해당 코드들은 `PyPy3`로 제출하면 통과하지만 `Python3`의 경우 시간 초과가 발생한다.😥