프로그래머스 문자열 내 마음대로 정렬하기 Swift

나의 풀이:

1.strings 중 n번째를 추려낸다.

2.n번쨰의 값들을 오름차순으로 정렬한다.

3.n번쨰의 값들 중 중복이 있는 지 확인한다.

4.중복이 있는 값들을 담는다.

5.n번째의 값이 중복이라면 strings 중 n번쨰가 중복인 것들을 찾아서 사전 순으로 정렬한 뒤 result에 넣어준다.

6.중복이 아니라면 오름차순한 배열을 바탕으로 strings 중 n번째를 가지고 있는 문자열을 찾아서 result에 넣어준다.

 

풀면서도 이렇게 푸는 건 아닐거라고 생각했다. 오지게 복잡하게 풀었기 때문이다.

import Foundation
 
func solution(_ strings:[String], _ n:Int) -> [String] {
    var result = [String]()
     var narr = [Character]()
    var overlap = [Character:Int]()
    let count = 1
    var overlapset:Set<Character> = []
    var sortoverlap = [String]()
 
1.strings 중 n번째를 추려낸다.
    for i in 0..<strings.count{
        narr.append(strings[i].map{$0}[n])
    }
2.n번쨰의 값들을 오름차순으로 정렬한다.
    let sortnarr = narr.sorted()
   
3.n번쨰의 값들 중 중복이 있는 지 확인한다.
 for i in 0..<sortnarr.count{
        if overlap[sortnarr[i]] != nil {
        overlap[sortnarr[i]] = count + 1
        } else{
        overlap[sortnarr[i]] = count
        }
    }
 
    let sortdic = overlap.keys.sorted()
   4.중복이 있는 값들을 담는다.
 for i in 0..<sortnarr.count {
        if overlap[sortnarr[i]]! > 1 {
            overlapset.insert(sortnarr[i])       }
    }
 
    for i in 0..<sortdic.count{
        sortoverlap.removeAll()
5.만약 중복이라면 strings 중 n번쨰가 중복인 것들을 찾아서 사전 순으로 정렬한 뒤 result에 넣어준다.
 
                if overlapset.contains(sortdic[i]) {
                    for j in 0..<strings.count{
                        let c = strings[j].map{$0}
                        if sortdic[i] == c[n] {
                            sortoverlap.append(strings[j])
                        }
 
                    }
                    sortoverlap.sort(by:<)
                    for j in 0..<sortoverlap.count{
                        result.append(sortoverlap[j])
                    }
6.만약 중복이 아니라면 오름차순한 배열을 바탕으로 strings 중 n번째를 가지고 있는 문자열을 찾아서 result에 넣어준다.
                }else{
                    for j in 0..<strings.count{
                        let c = strings[j].map{$0}
                        if sortdic[i] == c[n] {
                            result.append(strings[j])
                        }
                    }
        }
 
        }
 
    return result
}

다른 사람 중 가장 좋다고 생각하는 풀이

당연하게도 아주 쉽게 풀 수 있는 방법이 있었다. sorted안에 element들을 이용해 푸는건데 sorted는 알아서 element 2개를 넣어주고 조건을 넣어주면 알아서 정렬해준다.

1.문자열의 n번째를 알기 위해선 index의 위치를 알아야하는데 어차피 모든 문자열의 n번째는 똑같으므로 임의로 0번째 문자열의 0번째에서 n만큼 떨어진 곳의 index 즉 n번째의 index를 알아낸 뒤

2.result라는 상수를 만들고 strings에 sorted메소드를 적용한 뒤 element에 각 각 s1,s2를 넣어준 뒤 조건은 s1[index]와 s2[index]가 같을 때 즉, n번째 문자가 같을 땐 s1 과 s2를 비교해서 오름차순으로 정렬해라

n번째 문자가 같지 않을 때는 s1[index]와 s2[index]를 비교해서 오름차순으로 넣어라.

 

import Foundation
 
func solution(_ strings:[String], _ n:Int) -> [String] {
    let index = strings[0].index(strings[0].startIndex, offsetBy: n)
 
    let result = strings.sorted(by : {(s1 : String, s2 : String) -> Bool in
        if s1[index] == s2[index] {
            return s1 < s2
        }
        else {
            return s1[index] < s2[index]
        }
    })
 
    return result
}