프로그래머스 - 체육복(Swift)

프로그래머스 - 체육복(Swift)

 

 

 문제 설명

 

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

프로그래머스

 

 

 나의 풀이

 

여분이 있는 배열을 탐색하면서 잃어버린 배열의 값이 존재한다면 없애고, 앞의 번호가 존재한다면 앞에부터, 그 다음 뒤에 번호가 존재한다면 뒤에 번호에게 줘서 해결했지만 20개의 테스트케이스에서 1개의 테스트케이스를 통과하지 못했다.

 

고민 후 질문하기를 통해 5번 테스트케이스의 문제를 확인했는데 여분이 있는데 잃어버린 경우는 여분이 있는 배열을 탐색하기 전에 처리해줘야 한다는 것이었다. 

예를 들어 [5, 7]이 잃어버린 배열이고 [4, 5, 8]이 여분의 배열일 때 여분의 배열을 바로 탐색하면 4번이 5번의 체육복을 빌려주는 경우가 생기는 것이다. (원래는 잃어버린 번호와 여분의 번호가 같이 존재하는 경우 자기를 제외해야 한다)

 

문제를 잘 읽어보고 로직을 처리하는 능력이 부족했다. 

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

 

import Foundation

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
    
    var lostArr = lost.sorted(by: <)
    var reserveArr = reserve.sorted(by: <)
    
    reserveArr = reserveArr.filter{ 
        if lostArr.contains($0) {
            if let index = lostArr.firstIndex(of: $0) {
                lostArr.remove(at: index)
            }
            return false
        } else {
            return true
        }
    }
    
    for num in reserveArr {
        if lostArr.contains(num-1) {
            if let index = lostArr.firstIndex(of: num-1) {
                lostArr.remove(at: index)
            }
        } else if lostArr.contains(num+1) {
            if let index = lostArr.firstIndex(of: num+1) {
                lostArr.remove(at: index)
            }
        }
        
    }
    
    return n-lostArr.count
}

 

 

 피드백

 

1. 문제는 모든 문장이 문제이기 때문에 숨겨진 의미를 찾으려고 노력해야한다. 

2. 배열의 값을 지우는 것은 remove나 filter로 지울 수 있으며 remove를 이용할 때는 firstIndex로 찾고 if let바인딩을 통해서 지워야한다.