백준: 14425 문자열 집합

문제

14425번: 문자열 집합

 

문제 풀이

 N개에 문자열로 이루어진 집합 S가 있을 때, M개에 문자열 중 집합 S에 포함되는 문자가 총 몇 개인지 구하는 문제이다. 문제를 풀기 위해서는 `딕셔너리`와 `in`을 활용하면 쉽게 풀 수 있다.

 

 M개의 문자열 중, N에 포함되는지 확인하기 위해서는 `if pattern in strings`와 같이 확인하면 된다. 파이썬의 `in` 연산은 리스트의 경우 `O(N)`의 시간이 소요되지만 딕셔너리의 경우 `O(1)`의 시간이 소요되므로 보다 효율적이다.

 

코드

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()
        if pattern in strings:
            cnt += 1
    print(cnt)