[프로그래머스] 체육복

<문제>

---

<전략짜기>

우선 제한사항을 보면 전체 Input의 수는 크지 않아서 시간초과는 크게 문제가 될 것 같지는 않다. 도난당한 학생, 여벌의 체육복을 가진 학생이 각각 반드시 존재하니, 없는 경우에 대한 테스트 케이스는 따로 예외처리하지 않아도 될 것 같다. (보통 알고리즘 문제 풀때 테스트 케이스의 끝부분 조건(?) 에서 예외처리 해줘야 할 부분이 종종 생기는 것 같다.)

reserve를 기준으로 여벌이 있는 학생이 도난당한 학생(lost)에게 빌려준다고 상황을 가정하고 문제를 풀 것이다. 먼저 reserve에 있는 학생이 lost에도 있는지 체크 (여벌이 있어도 도난당했으면 빌려줄 수 없기 때문) 하고, 본인 앞 -> 뒤 순서로 lost에 포함되어 있으면 빌려주고, 도난당한 목록에서 지워주면 되겠다. 여기서 빌려주는 순서를 맞춰줘야 하는 이유는, 예를 들어서 n = 5, lost = [2, 4], reserve = [3, 5] 라고 주어졌을때 순서를 지키지 않으면 3번이 4번에게 대여를 해줘버리면, 5번이 있음에도 4번에게 대여를 해줄 수 없게 되기 때문이다. (+ 2번도 대여를 못받게 된다.)

전체학생수 - 도난당한 학생수 (lost) + 체육복을 빌린 학생수 로 결과를 계산해보려고 한다.

<코드>

import Foundation

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
    var result = n - lost.count
    var lost = lost.sorted()
    var reserve = reserve.sorted()
    
    for item in reserve {
        guard lost.count > 0 else { return result }    
        if lost.contains(item) {
            lost = lost.filter { $0 != item }
            result += 1
        } else if lost.contains(item-1) {
            lost = lost.filter { $0 != item-1 }
            result += 1
        } else if lost.contains(item+1) {
            lost = lost.filter { $0 != item+1 }
            result += 1
        }
    }
    return result
}

 

응..? 그런데 결과가..? 도대체 5번 테스트 케이스가 뭐길래..?

하.. 굉장히 긴시간동안 이런 당연한 예외케이스를 못찾아서 삽질을 했다 ㅠㅠ 😭 😭 😭 😭 😭

n = 5, lost = [2, 3, 4], reserve = [1, 2, 5] ...... 이런 케이스에서는 2번은 대여가 불가능해야 하는데 해버리게 된다....

if lost.contains(item) {
  lost = lost.filter { $0 != item }
  result += 1
}

이 부분에서 여벌이 있지만 도난당한 경우를 다 잡아줄거라고 쉽게 생각했는데, 아니였다 ㅎㅎ;

이런 현상이 발생한 이유는, reserve를 기준으로 빌려줄 대상을 찾으려고 해서 그런 것 같다. (idea thanks to 쇼디니 🤟 )

"나 여벌옷 있는데 너 빌릴래?"가 아니라.. "나 도난당했는데 너 여벌옷 있니?" 라고 접근했어야 했는데.. 지금 생각해보면 당연히 그렇게 접근 했어야 하는게 맞다. (학생으로 이입해서 생각해봐도 그렇다 ㅋㅋ) 암튼 기준이 잘못됐다고 생각이 들긴 했는데, 코드 고치기가 귀찮아서..

아예 도난당한 학생과 여벌이 있는 학생을 비교해서 중복되는 친구들은 모두 날리고 시작했더니 통과.. ㅋㅋ 👍

import Foundation

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
    var realLost = lost.filter { num -> Bool in
        return !reserve.contains(num)
    }.sorted()
    
    let realReserve = reserve.filter { num -> Bool in
        return !lost.contains(num)
    }.sorted()
    
    var result = n - realLost.count
    
    for item in realReserve {
        guard realLost.count > 0 else { return result }
        
        if realLost.contains(item-1) {
            realLost = realLost.filter { $0 != item-1 }
            result += 1
        } else if realLost.contains(item+1) {
            realLost = realLost.filter { $0 != item+1 }
            result += 1
        }
    }
    return result
}

<Lessons Learned>

문제 보고 음 그냥 하면 되겠는데? 쉬운데? 라고 접근했는데..ㅋㅋㅋㅋ 처음에 전략을 세울때 기준을 잡는 것에 대해 한번 더 신중히 생각해보고 해야겠다. 이러나 저러나 똑같겠지 했는데 이렇게 아닌 경우가 발생한다.. 시간으로 생각해볼 거리를 샀다고 치자 ... 😁 

 

알고리즘에 사용된 filter 내용이 궁금하다면?  👉  2020/12/15 - [iOS/기초를 탄탄히!] - [Swift] map, filter, reduce

 

 

출처 : 프로그래머스 체육복