Algorithm/프로그래머스

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

yanni13 2024. 3. 5. 15:56

 

 

 

사진 눌러서 문제 풀어보기

 

 

 

 

문제 설명

 

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

 

제한 조건

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

 

입출력 예

 

String n return
["sun", "bed", "car"] 1 ["car", "bed", "sun"]
["abce", "abcd", "cdx"] 2 ["abcd", "abce", "cdx"]

 

 

아이디어

 

 

  • 배열에 담긴 문자열의 갯수를 count 해서 그 개수만큼 for문 돌고
  • 단어마다 n번째 인덱스를 찾아 변수에 담은 후 변수끼리 비교하면 되겠다고 생각했다.
  • 인덱스가 같은 문자열일 경우에 대한 해답은 아직 생각하지 못했음

 

코드를 짜면서 계속 오류가 추가적으로 생겨났다 .. 그래서 시간이 지날수록 이렇게 푸는 거 맞나?라는 생각이 들었는데 아니었던 거 같다.

ㅋㅋ...

 

문제를 풀다가 시간이 너무 오래 걸려서 이틀째 생각하고 swift 배열 인덱스 조회하는 법 찾고 하다 보니 더 이상 시간을 지체할 수 없다는 생각에 다른 사람 코드를 참고해 보기로 하였다.

 

코드

func solution(_ strings:[String], _ n:Int) -> [String] {
    return strings.sorted{  Array($0)[n] == Array($1)[n] ? $0 < $1 :  Array($0)[n] < Array($1)[n] }
}

 

여러 사람들의 블로그를 참고하였지만 제일 간단하기도 하고 클로저를 잘 활용한 해답인거 같아 이 코드를 해석하면서 문제를 다시 풀어보았다.

 

이 코드를 보니 클로저를 알고 있어도 잘 활용하지 못하는 사람과 활용할 줄 아는 사람의 차이가 심하구나라는 생각이 들었다... ㅜㅜ

더 열심히 해야겠다.