본문 바로가기
📖 Coding Test/Programmers LV.0

[프로그래머스 LV.0] 배열의 유사도

by hyebin (Helia) 2023. 1. 13.
프로그래머스 LV.0 모음

배열의 유사도

문제 설명

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

제한 사항

  • 1 ≤ s1, s2의 길이 ≤ 100
  • 1 ≤ s1, s2의 원소의 길이 ≤ 10
  • s1과 s2의 원소는 알파벳 소문자로만 이루어져 있습니다
  • s1과 s2는 각각 중복된 원소를 갖지 않습니다.

입출력 예

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

입출력 예 설명

입출력 예 #1

  • "b"와 "c"가 같으므로 2를 return 합니다.

입출력 예 #2

  • 같은 원소가 없으므로 0을 return 합니다.

제출

import Foundation

func solution(_ s1:[String], _ s2:[String]) -> Int {
    return Set(s1).intersection(Set(s2)).count
}
두 배열을 집합 Set으로 변환해 두 집합의 교집합을 구한 후, 그 개수를 반환한다.

다른 풀이

import Foundation

func solution(_ s1:[String], _ s2:[String]) -> Int {
    return s1.filter{s2.contains($0)}.count
}
filter 함수를 사용해 s2 배열에 s1 배열의 요소가 포함되어 있는 경우의 개수를 구해 반환한다.
반응형

댓글