[JAVA] 내적

태로미·2023년 9월 6일
0

Programmers

목록 보기
1/29
post-thumbnail

❓문제


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


💻입출력 예


ab
[1,2,3,4][-3,-1,0,2]
[-1,0,1][1,0,-1]













📝해결


1. for문

class Solution {
  public int solution(int[] a, int[] b) {

    int sum = 0;

    for(int i=0; i<a.length; i++) {
      sum += a[i]*b[i];
    }

    return sum;
  }
}
  • for문으로 배열을 한 바퀴 돌면서 변수 sum에 각 배열의 인덱스값을 곱한 값을 누적합.
  • Stream을 사용할 수도 있지만 배열의 범위가 크지 않으므로 for문을 쓰는 것이 좀 더 효율적.

2. Stream_map

return IntStream.range(0, a.length)
				.map(index -> a[index] * b[index])
				.sum();
  • range() 메서드는 종료범위를 포함하지 않음.
    – index를 활용하는 것이므로 0부터 배열의 길이-1까지로 범위 설정.
  • map을 사용하여 a와 b의 각 인덱스를 곱한 값을 index에 저장, sum() 메서드로 더함.






💭마무리

  • Level_1 / 12:10
  • Point
    – 반복문을 사용한 누적합 구하기
    – Stream의 메서드 사용하기

처음 문제를 보자마자 Stream을 사용해서 풀어야겠다! 라고 생각해서 풀긴했는데
혹시 for문은 어떨까 싶어서 for문도 사용해봤다.
그런데 for문 코드가 좀 더 빨리 처리된다!😀
나름 배열 길이가 큰 것 같은데...이 정도쯤은 for문도 감당이 가능한듯ㅋㅋㅋ
앞으로도 이런 문제가 있으면 for문과 Stream 둘 다 사용해서
실행속도를 비교해봐야겠다~~😊


profile
Here And Now

0개의 댓글