특정값 존재여부 비교 - list와 set를 중심으로

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

10815번: 숫자 카드

 

제목을 굉장히 논문이름처럼 지었네.. 아무튼!!

문제

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.

셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다

출력

첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 가지고 있으면 1을, 아니면 0을 공백으로 구분해 출력한다.

 

N = int(input())
number = list(map(int,input().split()))
M = int(input())
check_number = list(map(int,input().split()))

for i in check_number:
  if i in number:
    print(1,end=" ")
  else:
    print(0,end=" ")

 

기존 작성한코드는 다음과 같다. 

N = int(input())
number = set(map(int, input().split()))  # 숫자들을 set으로 변환하여 저장
M = int(input())
check_number = list(map(int, input().split()))

for i in check_number:
    print(1 if i in number else 0, end=" ")

 

수정한 코드는 다음과 같다. 음? 체크할 넘버만 set으로 받았을 뿐인데 왜 시간초과가 해결된걸까?이건 파이썬에서의 리스트와 세트의 자료구조의 특징과 성능차이 때문이다.

리스트의 경우, 특정요소를 존재하는지 확인하려면 시작부터 끝까지 모든 요소를 확인해야한다. 이는 선현검색이라고 불리며, 최악의 경우 시간복잡도는 맨 끝에 요소가 존재하여, O(N)이 될것이다. 

세트의 경우, 해시 테이블을 이용하여 구현되어 있으며, 이로인해 특정 요소의 존재여부를 빠르게 확인할수 있다. 시간복잡도가 평균적으로 O(1)이다. 따라서, 많은수의 데이터중 특정값 존재 여부를 탐색하려면 set를 사용해야 하는것이다.