문제 설명
길이가 같은 두 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 입니다.
for문을 사용한 코드
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;
}
}
IntStream을 사용한 코드(다른 사람 코드)
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();
}
}- IntStream을 사용해 0부터 a.length까지의 인덱스를 포함하는 스트림을 생성함
- map()을 사용하여 각 인덱스에 있는 a배열과 b배열의 요소를 곱한 값을 새로운 스트림으로 변환함
- sum()을 사용하여 변환한 스트림의 모든 값을 더하여 최종 결과를 반환함
다른 사람 코드를 보다보면 Stream을 활용해서 풀이한 경우가 종종 보이는데,
Stream에 대해 잘 모르고 그로 인해 사용해보지 않아 항상 익숙한 반복문을 활용해 풀이하게 되는 것 같다.
Stream에 대해 더 공부해보고 사용해야겠다고 느꼈다.
댓글