상세 컨텐츠

본문 제목

Golang Algorithm - 특이한 정렬 (프로그래머스)

Programming Language/Go

by Yongari 2023. 7. 22. 11:58

본문

특이한 정렬
문제 설명
정수 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이 앞에 와야 합니다.

 

package main

import (
	"fmt"
	"math"
	"sort"
)

func solution(numlist []int, n int) []int {
	result := make([]int, len(numlist))
	copy(result, numlist)
	// fmt.Println(result)
	sort.Slice(result, func(i, j int) bool {
		numA := math.Abs(float64(result[i] - n))
		numB := math.Abs(float64(result[j] - n))
		fmt.Println(numA, numB)
		if numA == numB {
			return result[i] > result[j]
			// return result[i] < result[j]
		}

		return numA < numB
	})
	// fmt.Println(result)
	return result
}

func main() {
	numlist := []int{1, 2, 3, 4, 5, 6}
	n := 4
	solution(numlist, n)
}

 

 

 

관련글 더보기