1 minute read

가까운 수

문제 설명

정수 배열 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