프로그래머스 - 가까운 수 - C++

https://school.programmers.co.kr/learn/courses/30/lessons/120890

프로그래머스

 

#include <string>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;

int solution(vector<int> array, int n) {
    int answer = 0;
    vector<vector<int>> a (100,vector<int>());
    
    for(int i=0;i<array.size();i++)
    {
        a[abs(n-array[i])].push_back(array[i]);
    }
    
    for(int i=0;i<100;i++)
    {
        if(a[i].size()!=0)
        {
            sort(a[i].begin(),a[i].end());
            return a[i][0];
        }
    }
    
    return answer;
}

 

 

설명

abs(), sort(), 2차원 벡터 vector<vector<int>> 

array 배열에서 n 과 가장 가까운 수를 찾는 문제이다. 

원소가 1 에서 100까지니 차이는 0 ~ 99 까지이다. 

2차원 vector로  array[i] - n 의 절댓값이 i 일때  

a[abs(array[i] - n)].push_back(array[i]);

위 코드처럼 a[i] 에 array[i] 를 삽입하였다. 

예를 들어, array[i] 가 5 고 n 이 3 이면, array[i] - n 의 절댓값은 2 이다. 따라서 a[2] 에 5를 삽입한다. 

 

이런식으로 n과의 차이 즉, 거리에 따라 모든 숫자를 분류한다. 

그리고 a를 0부터 보며 a[i] 의 size가 1 이상인 것을 찾고 그 중 가장 작은 수를 찾으면 된다. 

 

a를 0부터 보는 이유는 a[0] 에는 n 과의 차이가 0인 즉, 제일 가까운 수가 들어있기 때문이다. 

 

ex) n이 5일 때, a[1] => { 6,4 } 이면 sort 를 통해 오름차순으로 정렬하고 a[1][0] 을 리턴하면

가장 가까운 수들 중 가장 작은 수를 찾을 수 있다.