https://school.programmers.co.kr/learn/courses/30/lessons/12932?language=java
[문제]
자연수 n을 뒤집어 각 자리 숫자를 원소로 가지는 배열 형태로 리턴해주세요. 예를 들어 n이 12345이면 [5,4,3,2,1]을 리턴합니다. n은 10,000,000,000 이하인 자연수입니다.
[입출력 예]
[풀이]
자연수 n이 주어지면 해당 숫자를 문자열로 변환합니다. n.toString()을 하면 결국 Char를 원소로 갖는 배열을 구성하게 됩니다. 그리고 해당 배열을 reverse 시키고, 다시 IntArray를 반환할 수 있도록 해주면 되겠습니다.
먼저 자바로 구현한다면
class Solution {
public static int[] solution(long n) {
//문자열로 변환하여 거꾸로 뒤집는다.
String reversedString = new StringBuilder(Long.toString(n)).reverse().toString();
int[] result = new int[reversedString.length()];
for (int i = 0; i < reversedString.length(); i++) {
//Char를 int형으로 변환하면서 저장
result[i] = Character.getNumericValue(reversedString.charAt(i));
}
return result;
}
}
- reverse() 함수를 통해 현재 문자열을 뒤집을 수 있습니다.
- 결과 값을 반환할 IntArray를 문자열 사이즈에 맞게 초기 선언합니다.
- 문자열의 사이즈 만큼 for문을 순회하면서 reverse 한 각 char를 int형으로 변환해서 result array에 넣어줍니다.
- 여기서 Character.getNumericValue은 특정 Char를 그에 해당하는 int형으로 변환해 줍니다.
만일 상응하는 int 값이 없다면 -1을 반환하게 됩니다.
이를 통해 result에는 [5, 4, 3, 2, 1]이 구성되는 것을 확인하실 수 있습니다.
코틀린으로 구현하는 것은 더 간단합니다.
fun solution(n: Long): IntArray =
n.toString().reversed().map { it.toString().toInt() }.toIntArray()
- 자바와 동일한 reverse 통해 치환된 문자열을 뒤집어줍니다.
- 이어서 각 요소를 바로 map 함수를 통해 int형으로 변환해줄 수 있습니다.
- map은 원본 컬렉션은 변경되지 않고 변환된 요소들로 이루어진 새로운 컬렉션을 반환해 줍니다.
- 그리고 곧바로 toIntArray() 함수를 통해 배열화시키면 완료입니다.
자바에 비해 코틀린은 주어진 컬렉션 함수들을 통해 더 간결하게 구현이 가능합니다.
자바 또한 getNumericValue()을 사용한 것처럼 다양한 함수의 활용도를 높여 풀어나가는 것이 좋아 보입니다.