[Swift] 프로그래머스 체육복(lv. 1)

[Swift] 프로그래머스 체육복(lv. 1)

 

1. 문제

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

프로그래머스

 

2. 접근

체육복 있냐고?

여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

이 두 가지 조건을 가볍게 보았다가 한 번 코드를 쓰고 엎었다.

여벌이 있어야만 체육복을 빌려줄 수 있다는 것... 만약 여벌이 있는데, 잃어버렸다면, 나는 내 것을 꼭 챙겨야 한다는 것...

(이 학교는 아마 일진이.... 없겠구나...)

 

아무튼 첫 번째 반복문에서는 innerReserve 즉, 여벌의 옷을 가진 사람이 옷을 잃어버렸는지 검사하는 로직이다. 만약, 여벌의 옷을 가진 사람이 옷을 잃어버렸다면, reserve 명단에서 제외시켜야 하기 때문에 먼저 돌려준다.

 

그다음 반복문은 구제받을 수 있는 학생들을 세주는 로직이다.

만약, 바로 앞이나 뒤의 학생에게만 옷을 빌릴 수 있기 때문에 조건문으로 먼저 걸러준다. 만약, 앞 뒤 학생이 있다면, reservedStudent 값을 하나 올려주고, innerReserve에서 옷을 빌려 준 학생을 삭제시킨다.

 

마지막으로 reservedStudent를 미리 구해놓은 answer과 더 해서 리턴한다.

 

3. 코드

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
    var answer = n - lost.count
    var innerLost = lost.sorted()
    var innerReserve = reserve.sorted()
    var reservedStudent = 0
    
    for i in innerLost {
        if innerReserve.contains(i) {
            innerLost.remove(at: innerLost.firstIndex(of: i)!)
            innerReserve.remove(at: innerReserve.firstIndex(of: i)!)
            answer += 1
        }
    }
    
    for i in innerLost {
        var k = 0
        for j in innerReserve {
            if i + 1 == j || i - 1 == j {
                reservedStudent += 1
                innerReserve.remove(at: k)
                break
            }
            k += 1
        }
    }
    return answer + reservedStudent
}