[swift] 프로그래머스 - 체육복

프로그래머스

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

문제 설명

 

- lost배열에는 체육복을 가져오지 못한 학생이, reserve배열에는 여분의 체육복을 가져온 학생이 입력됩니다.

 

- 여분의 체육복을 가져온 학생은 본인을 기준으로 앞, 뒤 학생에게 체육복을 빌려 줄 수 있습니다.

 

- lost배열과 reserve배열 둘 다 해당하는 학생도 존재하며, 해당 학생은 다른 학생에게 체육복을 빌려줄 수 없습니다.

 

- 체육복을 빌린 학생까지 포함 체육복이 있는 학생을 구합니다.

 

---

 

나의 풀이

import Foundation

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
    var lostStudent = lost.sorted()
    var reserveStudent = reserve.sorted()
    
    for i in reserveStudent {
        if lostStudent.contains(i) {
            lostStudent.remove(at: lostStudent.firstIndex(of: i)!)
            reserveStudent.remove(at: reserveStudent.firstIndex(of: i)!)
        }
    }
    
    for i in reserveStudent {
        guard !lostStudent.isEmpty else { break }
        
        if lostStudent.contains(i-1) {
            lostStudent.remove(at: lostStudent.firstIndex(of: i-1)!)
            continue
        }
        
        if lostStudent.contains(i+1) {
            lostStudent.remove(at: lostStudent.firstIndex(of: i+1)!)
            continue
        }
    }
    return n-lostStudent.count
}

 

lostStudent에는 체육복이 없는 학생이 있습니다.

체육복을 빌리게 된다면 해당 배열에서 지웠습니다.

따라서 n(모든학생) - lostStudent.count(체육복을 빌리지 못한 학생 수)로 값을 구합니다.

 

처음 주어지는 lost와 reserve배열은 정렬이 안되어 있기 때문에

정렬하여 변수에 담습니다.

 

lost와 reserve 양 쪽에 다 포함되는 값은

두 배열 모두에서 삭제시켜 줍니다.

 

여분이 있는 학생 배열을 순회하며

본인 앞, 뒤를 확인하며

체육복이 없는 학생이 해당되면

해당 학생은 lost배열에서 삭제되며

다음 반복으로 넘겨줍니다.

 

---

 

 

참고할만한 풀이

import Foundation

func solution(_ n: Int, _ lost: [Int], _ reserve: [Int]) -> Int {
    let newReserve = reserve.filter { !lost.contains($0) }
    let newLost = lost.filter { !reserve.contains($0) }

    var lostPeople: Int = newLost.count

    newReserve.forEach {
        let isLend: Bool = newLost.contains($0 - 1) || newLost.contains($0 + 1)
        if isLend && lostPeople > 0 {
            lostPeople -= 1
        }
    }

    return n - lostPeople
}

 

해당 풀이를 보면

풀이 방식은 비슷하지만

굳이 배열의 값을 지워가며 문제를 풀 필요가 없었습니다.

 

배열의 count값으로 체육복이 없는 학생 수를 파악해

체육복을 빌리면 -1씩 하여

체육복을 빌리지 못한 학생을 파악할 수 있었습니다.

 

또한 정렬을 할 필요가 없어서 시간면에서도 효율적입니다.

 

---