Buᐢ༝ᐢy 2023. 2. 16. 06:00

코딩테스트 연습 - 특이한 정렬

문제 설명


정수 n을 기준으로 n과 가까운 수부터 정렬하려고 합니다. 이때 n으로부터의 거리가 같다면 더 큰 수를 앞에 오도록 배치합니다. 정수가 담긴 배열 numlist와 정수 n이 주어질 때 numlist의 원소를 n으로부터 가까운 순서대로 정렬한 배열을 return하도록 solution 함수를 완성해주세요.

제한사항


  • 1 ≤ n ≤ 10,000
  • 1 ≤ numlist의 원소 ≤ 10,000
  • 1 ≤ numlist의 길이 ≤ 100
  • numlist는 중복된 원소를 갖지 않습니다.

입출력 예


numlist n result

[1, 2, 3, 4, 5, 6] 4 [4, 5, 3, 6, 2, 1]
[10000,20,36,47,40,6,10,7000] 30 [36, 40, 20, 47, 10, 6, 7000, 10000]

입출력 예 설명


입출력 예 #1

  • 4에서 가까운 순으로 [4, 5, 3, 6, 2, 1]을 return합니다.
  • 3과 5는 거리가 같으므로 더 큰 5가 앞에 와야 합니다.
  • 2와 6은 거리가 같으므로 더 큰 6이 앞에 와야 합니다.

입출력 예 #2

  • 30에서 가까운 순으로 [36, 40, 20, 47, 10, 6, 7000, 10000]을 return합니다.
  • 20과 40은 거리가 같으므로 더 큰 40이 앞에 와야 합니다.

제출 코드


2023년 1월 16일

#include <string>
#include <vector>
#include <algorithm>
#include <cstdlib>

using namespace std;

bool compare(pair<int, int>a, pair<int, int>b)
{
	if (a.first == b.first)
		return a.second > b.second;
	else
		return  a.first < b.first;
}

vector<int> solution(vector<int> numlist, int n) {
	vector<int> answer;
	vector<pair<int, int>> store;
	int length = numlist.size();

	for (int i = 0; i < length; i++)
	{
		store.push_back(make_pair(abs(numlist[i] - n), numlist[i]));
	}
	
	sort(store.begin(), store.end(),compare);

	int sLength = store.size();

	for (int i = 0; i < sLength; i++)
	{
		answer.push_back(store[i].second);
	}

	return answer;
}