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] }
}
여러 사람들의 블로그를 참고하였지만 제일 간단하기도 하고 클로저를 잘 활용한 해답인거 같아 이 코드를 해석하면서 문제를 다시 풀어보았다.
이 코드를 보니 클로저를 알고 있어도 잘 활용하지 못하는 사람과 활용할 줄 아는 사람의 차이가 심하구나라는 생각이 들었다... ㅜㅜ
더 열심히 해야겠다.