[Kotlin] 내적 (프로그래머스 알고리즘) : zip(), map(), .indices

 

 

 

 

문제

https://school.programmers.co.kr/learn/courses/30/lessons/70128

 

 

 

---

 

성공 코드

 

class Solution {
    fun solution(a: IntArray, b: IntArray): Int {
        val length = a.size - 1
        var sum = 0
        for(i in 0 .. length){
            sum += a[i] * b[i]
        }
        return sum
    }
}

 

---

 

설명

 

1. val length = a.size - 1
   a와 b는 같은 길이를 가지므로, 길이를 찾기위해 .size 프로퍼티를 이용한다.
   -1을 한 이유는 for문을 돌릴 때 size만큼 찾으면 마지막 위치의 배열이 없기 때문이다. (인덱스는 0부터 시작)
   -1을 하지 않고 for문에서 .. 대신 until을 써도 무방하다.
2. for(i in 0 .. length) { sum += a[i] * b[i] } 
   0부터 a.size - 1 까지 반복하면서 두 배열의 엘리멘트 곱을 시행한다.
   참고로 length에서 -1을 하지 않으면 ArrayIndexOutOfBoundsException 이 발생한다.
3. return sum
   이렇게 구한 sum을 리턴한다.

 

---

 

다른 방법

 

class Solution {
    fun solution(a: IntArray, b: IntArray): Int {
        return a.zip(b).map { it.first * it.second }.sum()
    }
}

• a.zip(b)
  두 개의 배열 a와 b를 순서대로 짝지어 페어(pair)로 묶는다.
  이렇게 하면 두 배열의 동일한 인덱스 위치에 있는 요소들끼리 짝지어진 리스트가 생성된다.
• map { it.first * it.second }
  각 페어에 대해서 첫 번째 요소와 두 번째 요소를 곱한 값을 계산하여 새로운 리스트를 생성한다. it은 현재 이터레이션에서의 페어를 나타낸다. it.first는 페어의 첫 번째 요소를, it.second는 페어의 두 번째 요소를 의미한다.
• sum()
  생성된 리스트의 모든 요소들을 더하여 총합을 계산한다.

 

💡 zip

val list1 = listOf("a", "b", "c")
val list2 = listOf(1, 2, 3)
val zipped = list1.zip(list2)
println(zipped) // 출력: [(a, 1), (b, 2), (c, 3)] 새로운 리스트 생성

• 두 개의 컬렉션을 쌍으로 묶어준다.
• 각 컬렉션에서 동일한 인덱스 위치에 있는 요소들끼리 페어(pair)를 만들어 새로운 리스트를 생성한다.

 

💡map

val numbers = listOf(1, 2, 3, 4, 5)
val squaredNumbers = numbers.map { it * it }
println(squaredNumbers) // 출력: [1, 4, 9, 16, 25]

• 각 요소를 주어진 람다 함수에 적용하여 새로운 컬렉션을 생성한다.
• 기존 컬렉션의 각 요소에 대해 주어진 람다 함수가 적용된 결과를 담은 새로운 컬렉션을 반환한다.

 

 

 

class Solution {
    fun solution(a: IntArray, b: IntArray): Int {
        var answer = 0
        for (i in a.indices)
            answer += (a[i] * b[i])
        return answer
    }
}

• 전체적으로 내 풀이와 비슷하나, for문에 범위에서 a.indices 를 이용하여 풀이했다.

 

💡.indices

val list = listOf("a", "b", "c")
for (i in list.indices) {
    println("Index $i: ${list[i]}")
}

•  컬렉션에 대한 인덱스를 나타내는 특별한 프로퍼티
• 컬렉션의 모든 요소에 대한 인덱스를 나타내는 범위(Range)를 제공
• 일반적으로 컬렉션의 인덱스를 반복하는 데 사용
  (for 루프에서 컬렉션의 인덱스를 반복하거나, 컬렉션의 일부 요소에 대한 작업을 수행할 때)

 

 

 

class Solution {
    fun solution(a: IntArray, b: IntArray): Int = a.mapIndexed { index, it -> it * b[index] }.sum()

• mapIndexed { index, it -> it * b[index] }
  mapIndexed 함수는 배열의 각 요소와 해당 인덱스에 대해 주어진 람다 함수를 적용한 결과를 새로운 리스트로 반환
  여기서는 a 배열의 각 요소 it과 해당 인덱스 index에 대해 it * b[index]를 계산하여 새로운 리스트를 생성
  즉, a 배열의 각 요소와 b 배열의 동일한 인덱스 위치에 있는 요소를 곱한 값을 리스트에 담음
• sum()
  생성된 리스트의 모든 요소들을 더하여 총합을 계산
• 람다함수의 용법을 보고 다시 복습해야 할 것 같다.