본문 바로가기

프로그래머스 코딩(자바)/Level 1

Programmers Level 1 - 음양 더하기

728x90

문제 설명

어떤 정수들이 있습니다. 이 정수들의 절댓값을 차례대로 담은 정수 배열 absolutes와 이 정수들의 부호를 차례대로 담은 불리언 배열 signs가 매개변수로 주어집니다. 실제 정수들의 합을 구하여 return 하도록 solution 함수를 완성해주세요.

 

제한사항
  • absolutes의 길이는 1 이상 1,000 이하입니다.
    • absolutes의 모든 수는 각각 1 이상 1,000 이하입니다.
  • signs의 길이는 absolutes의 길이와 같습니다.
    • signs[i] 가 참이면 absolutes[i] 의 실제 정수가 양수임을, 그렇지 않으면 음수임을 의미합니다.

 

입출력 예

absolutes signs result
[4,7,12] [true,false,true] 9
[1,2,3] [false,false,true] 0

입출력 예 설명

입출력 예 #1

  • signs가 [true,false,true] 이므로, 실제 수들의 값은 각각 4, -7, 12입니다.
  • 따라서 세 수의 합인 9를 return 해야 합니다.

입출력 예 #2

  • signs가 [false,false,true] 이므로, 실제 수들의 값은 각각 -1, -2, 3입니다.
  • 따라서 세 수의 합인 0을 return 해야 합니다.

 

1
2
3
4
5
6
7
8
9
class Solution {
    public int solution(int[] absolutes, boolean[] signs) {
        int answer = 0;
        for(int i=0;i<absolutes.length;i++) {
            answer += absolutes[i] * (signs[i] ? 1 : -1);
        }
        return answer;
    }
}
cs

 

  배열의 값을 더할때 signs에 따라 false면 -1을 곱하고 true이면 +1을 곱해 준다.

  1은 곱셈의 항등원이므로 값에는 변화가 없고 부호가 +,-로 변경된다.

 

1
2
3
4
5
6
7
8
9
10
import java.util.Arrays;
import java.util.concurrent.atomic.AtomicInteger;
class Solution {
    public int solution(int[] absolutes, boolean[] signs)  {
        AtomicInteger atomicInteger = new AtomicInteger(-1);
        return Arrays.stream(absolutes).reduce(0, (i, j) -> {
            return signs[atomicInteger.incrementAndGet()] ? i + j : i - j;
        });
    }
}
cs

 

  Stream.reduce(accumulator) 메서드는 Stream의 요소들을 하나의 데이터로 만드는 작업을 수행합니다.

  연산을 수행하는 부분은 accumulator 함수이며, 직접 구현해서 인자로 전달해야 합니다.

  Stream<Integer> numbers = Stream.of(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
  Optional<Integer> sum = numbers.reduce((x, y) -> x + y);
  sum.ifPresent(s -> System.out.println("sum: " + s));

  reduce()는 인자로 BinaryOperator 객체를 받는데, 
  BinaryOperator는 T 타입의 인자 두개를 받고 T 타입의 객체를 리턴하는 함수형 인터페이스입니다.

  BinaryOperator는 (total, n) -> total + n와 같은 형식으로 인자가 전달되는데요. 
  Stream의 1이 전달될 때, total(0) + n(1) = 1와 같이 계산되고, 
  여기서 리턴되는 1이 다음에 Stream에서 2가 전달될 때 total로 전달됩니다. 
  따라서, total(1) + n(2) = 3이 됩니다.
  다시 정리하면 아래와 같이 연산되면서, 마지막에는 1에서 10까지의 숫자를 더한 55가 리턴됩니다.
  total(0) + n(1) = 1
  total(1) + n(2) = 3
  total(3) + n(3) = 6
  total(6) + n(4) = 10
  total(10) + n(5) = 15
  total(15) + n(6) = 21
  total(21) + n(7) = 28
  total(28) + n(8) = 36
  total(36) + n(9) = 45
  total(45) + n(10) = 55

  위의 결과를 참조하면 부호의 배열값에 따라 +와 -를 수행하므로 원하는 결과가 나오게 됩니다.

  이때 sign 배열의 값을 얻어오기위한 인덱스를 증가 시키기 위해서 AtomicInteger를 사용하였습니다. 

  -1로 초기화하고 atomicInteger.incrementAndGet()을 이용하여 1을 증가시키고 값을 가져옵니다.

  
AtomicInteger는 int 자료형을 갖고 있는 wrapping 클래스입니다. AtomicInteger 클래스는 멀티쓰레드 환경에서 동시성을 보장합니다. AtomicInteger는 synchronized 보다 적은 비용으로 동시성을 보장할 수 있습니다.

 

728x90