[프로그래머스/Kotlin] 월간 코드 챌린지 시즌1 - 내적

문제

문제 출처

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

문제 설명

길이가 같은 두 1차원 정수 배열 a, b가 매개변수로 주어집니다. a와 b의 내적을 return 하도록 solution 함수를 완성해주세요.

이때, a와 b의 내적은 a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1] 입니다. (n은 a, b의 길이)

제한 사항

• a, b의 길이는 1 이상 1,000 이하입니다.
• a, b의 모든 수는 -1,000 이상 1,000 이하입니다.

입출력 예

a	b	reseult
[1,2,3,4]	[-3,-1,0,2]	3
[-1,0,1]	[1,0,-1]	-2

 

입출력 예 #1

• a와 b의 내적은 1*(-3) + 2*(-1) + 3*0 + 4*2 = 3 입니다.

입출력 예 #2

• a와 b의 내적은 (-1)*1 + 0*0 + 1*(-1) = -2 입니다.

 

풀이

a 배열과 b 배열의 첫 번째 원소부터 마지막 원소까지 곱하여 더한다.

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

for 문의 0..a.size-1의 경우 a.indices를 사용하여 간단히 표현할 수 있다.

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

 

다른 사람의 풀이 - 1

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

1. zip을 사용하면 두 리스트를 같은 index끼리 묶을 수 있다.
2. map 함수를 사용해 배열의 각 객체에 대한 연산을 반복한다.
3. sum 함수를 사용해 계산된 모든 값을 더한다.

 

다른 사람의 풀이 - 2

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

mapIndexed 함수는 map과 동일하지만 index까지 전달해준다. a 배열의 각 원소를 b 배열의 원소와 곱한 값으로 변경하고 sum 함수를 사용해 배열의 모든 값을 더한다.