본문 바로가기
Java/알고리즘

[프로그래머스/알고리즘] 내적 (자바/Java)

by 전재경 2022. 11. 19.

 

문제 설명

길이가 같은 두 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 이하입니다.

 

 

입출력 예 #1

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

입출력 예 #2

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

 

사실 ,,,, 내적 ....? 하면서 공식이 뭐지 어떻게 풀지 하면서 구글링을 했다..

근데 입출력 예를 보니 아,,, 이말이구나 싶었다 ㅋㅋㅋㅋ

 

Step1. 문제분석

내적 이라는 말 때문에 어떤 공식을 사용해야 하는지 찾았는데 입출력 예를 보니 바로 이해가 됐다.

배열의 첫번째 끼리, 두번째 끼리, 각각 곱해서 더하면 되는 너무도 간단한 문제였다.

for 문을 통해 길이만큼 돌고 그 길이만큼 배열의 수를 곱해서 return 해주면 되는 것이였다.

 

Step2. 문제풀이 코드

class Solution {
    public int solution(int[] a, int[] b) {
        int answer = 0;
        
        for(int i = 0; i < a.length; i++){
            answer += a[i] * b[i];
        }
        return answer;
    }
}

짜란... 너무도 간단

 

Step3. 다른 사람의 풀이 코드

import java.util.stream.IntStream;

class Solution {
    public int solution(int[] a, int[] b) {
        return IntStream.range(0, a.length).map(index -> a[index] * b[index]).sum();
    }
}

stream으로 풀이를 하는 경우가 참 많은거 같다 ...

반드시 공부해서 내것으로 만들어야겠다 !!

댓글