[프로그래머스]가까운 수
가까운 수
문제 설명
정수 배열 array와 정수 n이 매개변수로 주어질 때, array에 들어있는 정수 중 n과 가장 가까운 수를 return 하도록 solution 함수를 완성해주세요.
제한사항
- 1 ≤ array의 길이 ≤ 100
- 1 ≤ array의 원소 ≤ 100
- 1 ≤ n ≤ 100
- 가장 가까운 수가 여러 개일 경우 더 작은 수를 return 합니다.
입출력 예
array | n | result |
---|---|---|
[3, 10, 28] | 20 | 28 |
[10, 11, 12] | 13 | 12 |
입출력 예 설명
입출력 예 #1
- 3, 10, 28 중 20과 가장 가까운 수는 28입니다.
입출력 예 #2
- 10, 11, 12 중 13과 가장 가까운 수는 12입니다.
나의 풀이 코드
import java.util.*;
class Solution {
public int solution(int[] array, int n) {
int answer = 0;
List<Integer> dif_list = new ArrayList<>(); //각 요소와 n과의 차이 리스트
List<Integer> min_list = new ArrayList<>(); //가장 가까운 요소들만 담긴 리스트
int min = 0;
for (int i=0; i<array.length; i++){//배열의 각요소과 n과의 차이를 저장
dif_list.add(Math.abs(array[i] - n));
}
min = Collections.min(dif_list);//차이가 가장 작은 값을 구함.
for (int i=0; i<array.length; i++){//가장 작은 값과 같은 값의 요소만 필터링
if (dif_list.get(i).equals(min))
min_list.add(array[i]);
}
answer = Collections.min(min_list);//요소들중 더 작은 수 return
return answer;
}
}
배운점
import java.util.Arrays;
import java.util.stream.Collectors;
class Solution {
public int solution(int[] array, int n) {
return Math.min( // 차이가 최소값이 두개인경우 작은 값 반환
array[Arrays.stream(array)
.map(operand -> Math.abs(n - operand)) //차이 계산 후 각 요소와 매핑
.boxed()
.collect(Collectors.toList())
.indexOf( // 아래 연산 후 나온 인덱스 찾기
Arrays.stream(array)
.map(operand -> Math.abs(n - operand))
.min()
.orElse(0)
)],
array[Arrays.stream(array) //위 코드와 동일
.map(operand -> Math.abs(n - operand))
.boxed()
.collect(Collectors.toList())
.lastIndexOf(
Arrays.stream(array)
.map(operand -> Math.abs(n - operand))
.min()
.orElse(0)
)]
);
}
}
다른 사람이 스트림으로 푼 코드이다. 주석에 설명을 달았다.
Leave a comment