본문 바로가기

Python

[Python/level 0] 배열의 유사도 - 120903, len, 리스트 컴프리헨션

문제 설명

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

 

code - 1차

def solution(s1, s2):
    count = 0
    s1.sort()
    s2.sort()
    if len(s1) >= len(s2):
        for i in s1:
            if i in s2:
                count += 1
    if len(s1) < len(s2):
        for i in s2:
            if i in s1:
                count += 1
    return count

 

code - 2차

def solution(s1, s2):
    return len([s for s in s1 if s in s2])
  • 코드 리팩토링 내역
    • 불필요한 sort를 제거하였습니다.
    • 조건에 맞는 s를 list로 받아, len 길이를 구하였습니다.

 

 

반응형