프로그래머스_Python_집합 set() 활용해서 교집합 풀기

1. 프로그래머스 python 코딩테스트 입문

Level 0. 배열의 유사도

두 배열이 얼마나 유사한지 확인해보려고 합니다. 문자열 배열 s1과 s2가 주어질 때 같은 원소의 개수를 return하도록 solution 함수를 완성해주세요.

s1 s2 result
["a", "b", "c"]["com", "b", "d", "p", "c"] 2
["n", "omg"]["m", "dot"] 0

정답 확인

평범한 방법

def solution(s1, s2):
    answer = 0
    for i in s1:
        if i in s2:
            answer += 1
    return answer

정답 확인 (파이썬 집합 set() 활용)

등차수열 n(n+1)/2

def solution(s1, s2):
    return len(set(s1)&set(s2))

• set() : 리스트나 문자열 등을 집합으로 바꿔줌.
• 집합 : 중복제거 & 순서 없음.
• in검사, 교집합/합집합 등에 사용.
• & 연산자 : 두 집합의 교집합
• set(s1) & set(s2) : 두 리스트의 교집합
• len() : 집합의 길이 = 공통 원소 개수