[Algorithm] 프로그래머스 - 배열의 유사도 (Kotlin)

두 배열이 얼마나 유사한지 확인하려고 한다. 문자열 배열 s1, s2가 주어질 때 같은 원소 개수를 리턴하는 solution()을 완성하라

 

 

2중 for문을 써서 주먹구구식으로 풀면 아래와 같다.

 

class Solution {
    fun solution(s1: Array<String>, s2: Array<String>): Int {
        var count = 0
        for (i in s1) {
            for (j in s2) {
                if (i == j) {
                    count++
                }
            }
        }

        return count
    }
}

 

이것을 간결하게 만들려고 시도했는데 잘 안 되서 다른 사람 풀이를 참고했다.

 

class Solution {
    fun solution(s1: Array<String>, s2: Array<String>) = s1.map(s2::contains).count { it }
}

 

또는 아래처럼도 할 수 있다.

 

class Solution {
    fun solution(s1: Array<String>, s2: Array<String>): Int = s1.count { s2.contains(it) }
}