[프로그래머스] (Swift) Kakao - [3차]파일명 정렬 (Lv.3) (feat. 예외 케이스 적용시켜 보기)

🟠 TIL 오늘 배운 것

📝 클로저 활용해서 array 이중(?), 다중 정렬하기

클로저를 사용해서 array에 여러 조건별로 정렬해줄 수 있다. 이 방법에는, 아래와 같은 방법이 있다.

let myGrades = [
    ["B", "다"],
    ["A", "바"],
    ["C", "나"],
    ["B", "가"],
    ["A", "라"],
    ["A", "마"]
]

print(myGrades.sorted { ($0[0] < $1[0]) || ($0[1] > $1[1]) })
// [["A", "바"], ["A", "마"], ["A", "라"], ["B", "다"], ["B", "가"], ["C", "나"]]

이렇게 하게되면, ABC 오름차순은 그대로 유지하면서, 뒤에 '가나다'는 내림차순으로 정렬했다. || 논리연산자를 통해 그 기준을 유지하고 뒤에 추가 조건을 줄 수 있다. 

원래 이렇게 하나의 클로저 내부에서 sorted by 를 이용해서 정렬했었는데, 다른 방법이 있었다. 신박했다!

let customSort: ([String], [String]) -> Bool = { a, b in
    if a[0].lowercased() != b[0].lowercased() {
        return a[0].lowercased() < b[0].lowercased() // 다르다면 사전순정렬
    } else {
        if let n1 = Int(a[1]), let n2 = Int(b[1]), n1 != n2 {
            return n1 < n2 // 숫자다르다면 큰순으로 정렬
        }
    }
    return false // 문자도 같고 숫자도 같다면 안바꿈
}

newFiles.sort(by: customSort)

이렇게 정렬 클로저를 사용하는 방식이다. by: 안에 연산자를 넣어야할 것같지만, 연산프로퍼티(?맞나요?)를 대신해서 넣었다. sorted(by:) 함수는 인자로 전달받은 customSort(_:_:)의 규칙을 따라 어레이를 정렬하고 리턴해준다.

정렬할때, 이렇게 조건을 이용해서 기준을 만들어주고 그 규규칙을 따르게끔 정렬해주는 방식도 있다는 것을 꼭 기억해야겠다. 위처럼 연산프로퍼티가 아니라 함수로도 정렬 클로저 규칙을 생성해서 넣어줄 수 있다.

let names = ["Chris", "Alex", "Ewa", "Barry", "Daniella"]

func backward(_ s1: String, _ s2: String) -> Bool {
    return s1 > s2
}
var reversedNames = names.sorted(by: backward)
// reversedNames is equal to ["Ewa", "Daniella", "Chris", "Barry", "Alex"]

이렇게 하게 되도, by안에서 해당 함수의 규칙을 따라 값을 리턴해준다.

 

📝 map을 이용해서 Array에 있는 원소들을 합쳐서 출력하기

원래 나는 for문을 이용해서 출력해주었다. 근데 이거... 고차함수 map으로 응용이 가능하다.

원래대로라면, [["a","b","c"], ["d","e","f"]] 를 ["abc", "def"]로 출력하기 위해서 아래와 같은 방식으로 진행했다.

for f in sortedFiles {
    var temp = ""
    for i in 0..<f.count {
        temp += f[i]
    }
    answer.append(temp)
}

for문으로 한 [array]씩 꺼낸 후, 거기서 또 길이만큼 for 문을 돌려서 더해주었다. 무려 이중포문..을사용했다.

하지만 이를 고차함수 map 과 join 이라는 메서드를 이용하면 간단하게 출력 가능하다.

newFiles.map{$0.joined()}

newfile에 있는 array들을 한덩이씩 꺼낸 후 ($0), 여기있는 것들을 string 으로 전부 엮어주는 join 메서드를 사용했다. 간단... 스.. 하다. .. 앞으로 이런 일이 있을때 map 과 join 을 꼭 기억해내자!

 

🟠 문제링크

https://school.programmers.co.kr/learn/courses/30/lessons/17686

프로그래머스

 

🟠 내가 푼 풀이

file 문자열 나누기

• 우선 주어진 file들이 head, number, tail 에 따라서 정렬 순서가 달라지므로, file의 문자열을 분리해야한다고 생각했다.
• isnumber, isletter을 이용해서 split 해주었다.
  • isLetter는 영어 대소문자를 의미하는 듯했다. 온점이나 하이픈, 띄어쓰기 (공백)은 인식하지 못해서, 따로 처리해주어야햇다. 
  • isASCII도 있었는데 이는 문자 전체를 지칭하고 있어서 letter도, 온점 공백 등등도 모두 해당하는 조건이라 사용할 수 없었다.
  • 어차피 파일명에 들어가는 것이 온점, 하이픈(-), 공백뿐이라서 세개를 이용해서 따로 배열을 만들어주고, 여기에 포함하는지에 대한 여부로 조건처리를 진행했다.
• istail이라는 변수를 주었다.
  • 숫자를 기준으로 앞이 head, 뒤가 tail 인데, 이를 구별하기 위해서 bool값을 하나 준 것이다.
  • 만약 istail 이 true값을 가지면, 이미 중간지점을 지났다는 뜻이므로, 그 이후에 만나는 문자들은 모두 tail에 넣어주는 방식으로 진행했다.

정렬하기

정렬 코드는 인터넷을 참고했다. 원래 아래와 같은 방식으로 진행했는데, 대문자가 먼저 정렬되거나, 두번째 조건때문에 앞 조건이 유지되지 않은 채로 두번재 조건으로 정렬이 되어버리는 경우가 생겼다.

var sortedFiles = newFiles.sorted(by: { $0[0] < $1[0] })
sortedFiles = sortedFiles.sorted(by: { Int($0[1])! < Int($1[1])! })

맞다. 조건을 두번 따로 주었으니까, 첫번째 정렬한 것이 두번재까지 유지되지 않은 채로 따로 정렬됐던 것이다. 그리고 대소문자를 따로 구분하지 않는다는 조건이 나와있는데, 나는 그냥 정렬해버렸기 때문에 대소문자에 대한 우선순위도 고려한채로 정렬이 진행됐다. (a와 A를 비교함, 스위프트는 대문자가 소문자보다 작은것으로 간주한다.)

그래서 아래와 같은 코드를 참고했다.

let customSort: ([String], [String]) -> Bool = { a, b in
    if a[0].lowercased() != b[0].lowercased() {
        return a[0].lowercased() < b[0].lowercased() // 다르다면 사전순정렬
    } else {
        if let n1 = Int(a[1]), let n2 = Int(b[1]), n1 != n2 {
            return n1 < n2 // 숫자다르다면 큰순으로 정렬
        }
    }
    return false // 문자도 같고 숫자도 같다면 안바꿈
}

newFiles.sort(by: customSort)

문제 조건에서 대소문자의 구분은 없다고 했기 때문에, a와 A를 비교하지 않고 lowercase로 변형시켜서 비교해주었다. 이렇게 하게 되면 A를 잠깐 a로 바꿔서 비교하게 되므로, 두개에 동일한 우선순위가 부여된다. 

그리고 문자의 정렬된 순서가 같다면, number측에서 정렬을 진행한다. 숫자가 다르다면 큰순으로 정렬해주었고, 마지막으로 if문을 통과하지 않은 것들에 대해서는 문자도 같고 숫자도 같다는 것이기 때문에 false를 반환하면서 아무련 정렬도 해주지 않았다.

 

🟠 첫번째 시도 (실패 9개, 55점)

코드는 더보기를 클릭해보자.

틀린 코드

func solution(_ files:[String]) -> [String] {
    var answer = [String]()
    var newFiles = splitFiles(files)
    
    let customSort: ([String], [String]) -> Bool = { a, b in
        if a[0].lowercased() != b[0].lowercased() {
            return a[0].lowercased() < b[0].lowercased() // 다르다면 사전순정렬
        } else {
            if let n1 = Int(a[1]), let n2 = Int(b[1]), n1 != n2 {
                return n1 < n2 // 숫자다르다면 큰순으로 정렬
            }
        }
        return false // 문자도 같고 숫자도 같다면 안바꿈
    }

    newFiles.sort(by: customSort)
    return newFiles.map{$0.joined()}
}

func splitFiles(_ files: [String]) -> [[String]] {
    let symbol = [".", "-", " "]
    var splitFiles = [[String]]()
    for f in files {
        var head = ""
        var num = ""
        var tail = ""
        var istail = false
        for char in f {
            if (char.isLetter && istail) || (istail && symbol.contains(String(char))) {
                tail += String(char)
            } else if char.isNumber {
                num += String(char)
                istail = true
            } else if (!istail && char.isLetter) || (!istail && symbol.contains(String(char))) {
                head += String(char)
            }
        }
        splitFiles.append([head, num, tail])
    }
    return splitFiles
}

print(solution(files))

결과는,, ㅊ... 충격..

왜틀렸는지 알아보쟈... 

 

🟠 두번째 시도 (정답!)

왜 틀렸을까!! 런타임에러도, 시간초과도 아니고 실패인것으로 보아, 몇개의 예외케이스에서 정답을 잘 리턴하지 못했을 것이다. 어떤 예외사항이 있을지 '프로그래머스 질문하기'를 뒤지면서, 여러 사람들의 에러사항들을 보면서 원인을 찾아나섰다. (이거 이렇게 예외케이스를 질문하기, 이런거 없으면 어떻게 생각해내죠? ㅠㅠ 시험볼때 이런 예외케이스를 어떻게 생각하냔 말이댜 ㅠㅠ)

여러사람들의 케이스를 살펴보다가, '숫자로 끝나는 파일명'일 경우에, 숫자를 제대로 tail에 넣지 못하는 이슈가 있었다고 한다. 바로 케이스를 적용해보았다.

let files = ["O00321", "O49qcGPHuRLR5FEfoO00321"]

결과가 위처럼 나왔다. head는 영어 O가 잘 들어가지만, 두번째 케이스 number에 주목해보자. 49만 number에 들어가고, qcGPHuRLR5FEfoO00321는 전부 tail 에 들어가야하는데, tail 부분에 들어가야할 숫자들이 전부 number에 들어가있다.

내가 짠 조건을 살펴보자.

if (char.isLetter && istail) || (istail && symbol.contains(String(char))) {
    tail += String(char)
} else if char.isNumber {
    num += String(char)
    istail = true
} else if (!istail && char.isLetter) || (!istail && symbol.contains(String(char))) {
    head += String(char)
}

isNumber에 해당하면 무조건 num string에다가 추가시켜주는것이 문제였다. 그럼 tail부분에 들어갈 숫자들을 어떻게 구별해주면 좋을까? 

파일명은 '숫자'로 head와 tail로 구별된다. 그래서 A12B34 가 있다고 칠때, 12를 기준으로 A는 head, B34는 tail 로 가야한다. 뒤에 3과 4는 그러면, 숫자이면서, tail 이 비어있지 않으면, tail 에 넣으면된다. 1과 2는 숫자이면서, tail 이 비어있을 경우, num에다가 추가해주면 될 것이다. 조건을 아래와 같이 바꾸어서 진행해보니, 성공했다 ❤️‍🔥

if (char.isLetter && istail) || (istail && symbol.contains(String(char))) || (!tail.isEmpty && char.isNumber && istail) {
    tail += String(char)
} else if char.isNumber && tail.isEmpty {
    num += String(char)
    istail = true
} else if (!istail && char.isLetter) || (!istail && symbol.contains(String(char))) {
    head += String(char)
}

휴,, 비교적 오류를 빨리찾아서 약 30분만에 해결했다. 도대체... 코테볼때 예외케이스를 어떻게 생각해낵냐고오 ㅠㅠㅠ 너무어렵다.

 

🟠 정답코드

https://github.com/deslog/Algorithm/blob/main/Algorithm/Programmers/kakao_%ED%8C%8C%EC%9D%BC%EB%AA%85%20%EC%A0%95%EB%A0%AC/main.swift

GitHub - deslog/Algorithm: ✨ 백준, 프로그래머스 Swift 문제풀이 ✨

func solution(_ files:[String]) -> [String] {
    var newFiles = splitFiles(files)

    let customSort: ([String], [String]) -> Bool = { a, b in
        if a[0].lowercased() != b[0].lowercased() {
            return a[0].lowercased() < b[0].lowercased()
        } else {
            if let n1 = Int(a[1]), let n2 = Int(b[1]), n1 != n2 {
                return n1 < n2 
            }
        }
        return false
    }

    newFiles.sort(by: customSort)
    return newFiles.map{$0.joined()}
}

func splitFiles(_ files: [String]) -> [[String]] {
    let symbol = [".", "-", " "]
    var splitFiles = [[String]]()
    for f in files {
        var head = ""
        var num = ""
        var tail = ""
        var istail = false
        for char in f {
            if (char.isLetter && istail) || (istail && symbol.contains(String(char))) || (!tail.isEmpty && char.isNumber && istail) {
                tail += String(char)
            } else if char.isNumber && tail.isEmpty {
                num += String(char)
                istail = true
            } else if (!istail && char.isLetter) || (!istail && symbol.contains(String(char))) {
                head += String(char)
            }
        }
        print("head은 \(head), number은 \(num), tali 은 \(tail)")
        splitFiles.append([head, num, tail])
    }
    return splitFiles
}