[python] 파이썬 리스트 비교하기 (set 자료형 이용하기!)

들어가며

코딩테스트를 연습하고 다른사람들의 풀이를 공부하다 보면 굉장히 기초적인 부분에서 코드의 길이를 줄이고, 시간복잡도를 줄일 수 있다는 것을 느낀다. 따라서 코딩테스트에서 헷갈리지 않고 능숙하게 사용할 수 있도록 set 집합에 대해서 정리해두자!

 

1. 파이썬 리스트 중복 제거하기

a = [1,2,3,3,3,3,4,5,6,7,3,5,6,7]
a = set(a)
print(a)

# 다시 리스트로 사용하고 싶다면?
a = list(a)
a

set은 집합형태의 자료형이다. 따라서 중복되는 값이 하나의 집합에 들어갈 수 없는 특성을 사용해서 리스트 내의 중복값을 간단하게 제외해줄 수 있다.

 

2. 파이썬 리스트 비교하기 (교집합, 합집합, 차집합을 이용하자)

2.1 교집합

a = [1,2,3,4,5]
b = [2,4,999,8,2,1]

intersection = set(a) & set(b)
print(list(intersection))

두 개의 교집합!! 코딩테스트에서 '복권 당첨됐는지 확인하기' 등 두 리스트를 비교할 때 해당 방법을 유용하게 사용한다. 결괏값은 list로 바꿔줘 보았다.

 

2.2 합집합

a = [1,2,3,4,5]
b = [2,4,999,8,2,1]

union = set(a) | set(b)
print(union)

 

2.3 차집합

a = [1,2,3,4,5]
b = [2,4,999,8,2,1]

differ_a = set(a) - set(b)
differ_b = set(b) - set(a)
print(differ_a)
print(differ_b)

이 차집합 때문에 해당 포스팅을 작성했다. 이 문제를 풀때 나는 if문을 이용해서 코드를 길게 썼지만, set의 차집합을 이용하면 단번에 한줄로 구할 수 있다는 것을 알게 되었다. 차집합도 코딩테스트를 풀때 아주 유용하게, 자주 사용되는 것 같다!

2.4 대칭 차집합

위에 차집합은 a 기준에서 b를 빼거나 b 기준에서 a를 빼는 연산을 수행한다. 하지만 두 리스트에서 겹치는 값을 제외한 '모든 값'을 구해야할 때도 있지 않은가? 이떄는 바로 대칭 차집합을 이용하면된다.

a = [1,2,3,4,5]
b = [2,4,999,8,2,1]

differ_all = set(a) ^ set(b)
print(differ_all)

위에 차집합들의 값이 한 set로 합쳐져서 나온다.