[프로그래머스/Kotlin] 문자열 내 마음대로 정렬하기

문제 링크 -> https://programmers.co.kr/learn/courses/30/lessons/12915

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

제한 조건

• strings는 길이 1 이상, 50이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

------------------------------------------------------------풀이------------------------------------------------------------------
* 세 번째로 작성한 코드만 보시면 됩니다!
1. sortedWith을 사용하고 comaparator를 지정해서 정렬한후 TypedArray로 형변환해서 return합니다.

/////////////////////////처음 작성한 코드 
class Solutions {
    fun solution(strings: Array<String>, n: Int): Array<String> {
        var pairList = ArrayList<Pair<String,String>>(strings.size)

        for( i in strings){
            val pair = Pair<String, String>( i[n].toString(), i)
            pairList.add(pair)
        }

        return pairList.sortedWith( compareBy({it.first}, {it.second})).map{ it.second}.toTypedArray()
    }
}

//////////////////////두 번째로 작성한 코드
class Solution {
    fun solution(strings: Array<String>, n: Int): Array<String> =
        strings.map{ (it[n].toString() to it) }.sortedWith( compareBy({it.first}, {it.second})).map{ it.second}.toTypedArray()
}

/////////////////////세 번째로 작성한 코드
class Solution {
    fun solution(strings: Array<String>, n: Int): Array<String> =
        strings.sortedWith(compareBy({it[n]}, {it})).toTypedArray()
}

왜 Pair에 집착했을까요..그냥 it으로 바로 comparator지정해주면 되는데 ㅜㅜ 

------------------------------------------------------------결과------------------------------------------------------------------

위 3개의 코드는 전푸 통과한 풀이입니다! 

 

 

 

 

 

- 문제에 대한 질문 댓글 환영!
- 중간에 잘못된 부분이 있다면 댓글로 남겨주세요. 수정하겠습니다.