[n14425] 문자열 집합 in python

---

문자열 문제!

아래 코드는 시간초과가 떴다.

그래서 리스트 대신 set 자료구조를 이용해서

돌려보았더니 시간 초과 문제가 해결되었다!

 

저번 문제에서도 느꼈지만

다루는 데이터의 수가 커질수록

리스트의 연산시간은 길어진다.

(이럴 땐 사용 지양)

 

import sys
input=sys.stdin.readline

n, m = map(int, input().split())
sArr = [input() for _ in range(n)]
cnt=0
for _ in range(m):
    temp = input()
    for i in range(n):
        if temp == sArr[i]:
            cnt+=1
print(cnt)

 

아래 코드는 시간초과 x

무난하게 통과하였다!

 

import sys
input=sys.stdin.readline

n, m = map(int, input().split())
sArr = set([input() for _ in range(n)])
cnt=0
for _ in range(m):
    temp = input()
    if temp in sArr:
        cnt+=1
print(cnt)