Algorithm/Programmers

[프로그래머스] 자릿수 더하기 - Java

뭐든 해보기 2023. 4. 4. 17:00

문제) 정수 n이 매개변수로 주어질 때 n의 각 자리 숫자의 합을 return하도록 solution 함수를 완성해주세요

 

제한사항)

  • 0 ≤ n ≤ 1,000,000

 

입출력 예)

n result
1234 10
930211 16

 

입출력 예 설명)

입출력 예 #1

  • 1 + 2 + 3 + 4 = 10을 return합니다.

입출력 예 #2

  • 9 + 3 + 0 + 2 + 1 + 1 = 16을 return합니다.

풀이)

1. Stream을 사용하여 int 배열로 변환

2. Arrays stream를 사용한 합계 출력

import java.util.Arrays;
import java.util.stream.Stream;
class Solution {
    public int solution(int n) {
        	int[] digits = Stream.of(String.valueOf(n).split("")).mapToInt(Integer::parseInt).toArray();
        return Arrays.stream(digits).sum();
    }
}

 

위처럼 배열로 선언하지 않고 한꺼번에 쓴다면 복잡하지만 한줄로 나타낼 수 있다.

Arrays.stream(String.valueOf(n).split("")).mapToInt(Integer::parseInt).sum();

stream이 쓰여지는 방향이 무궁무진 한듯..