프로그래머스 체육복 Swift

점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있습니다. 예를 들어, 4번 학생은 3번 학생이나 5번 학생에게만 체육복을 빌려줄 수 있습니다. 체육복이 없으면 수업을 들을 수 없기 때문에 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 들어야 합니다.

전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성해주세요.

제한사항

• 전체 학생의 수는 2명 이상 30명 이하입니다.
• 체육복을 도난당한 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
• 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

입출력 예

nlostreservereturn

n	lost	reserve	return
5	[2, 4]	[1, 3, 5]	5
5	[2, 4]	[3]	4
3	[3]	[1]	2

https://roeldowney.tistory.com/91 이 분의 코드를 참조했다.

[Swift]프로그래머스 체육복

풀이:

//도둑맞은 학생들과 여분이 있는 학생들의 배열을 또 다른 배열을 만들어 각 각 담는다 -> 파라미터 안의 배열은 상수로 되어있기 때문에 값을 바꿀 수가 없다.

//학생 수

그리고 학생 숫자를 셀 변수도 하나 만들어준다.(studentnumber) 학생 숫자를 세는 변수엔 우선은 잃어버린 학생 수를 빼서 시작해준다.

 

//여분이 있는 학생이자 동시에 도둑맞은 학생

맨 처음 확인해 주어야 할 건 도둑 맞은 학생 중 여분이 있는 학생이 있는 가를 찾아야 한다. 왜냐하면 다음 for문부터는 여분이 있는 학생 중 -1 또는 +1자리 중 도둑 맞은 학생이 있는가를 찾을 것 이기 때문에 잘못하면 자신도 도둑맞았는데 여분이 있는 체육복을 다른 도둑 맞은 학생에게 줄 수 있기 때문이다. 그리고 학생 수에 +1을 해준다. 도둑 맞았지만 어차피 여분이 있으니 수업에 참가할 수 있기 때문이다.

//받은 학생 또 받는 거 방지

그리고 또 확인해야할 건 받았는데 또 받은 학생이 있을 수 있다. 예를 들어 2,4번 학생이 도둑을 맞고 1번,3번 학생이 여분이 있는데 1번 학생이 2번 학생에게 주었는데 또 3번 학생이 2번 학생에게 줄 수 있는 것을 막아야 한다.

고로 도둑맞은 학생의 배열 중에서 여분이 있는 학생의 배열과 같으면 그 해당 인덱스의 숫자를 0으로 만들어준다.

숫자가 0이라는 건 주었거나 또는 받았다는 걸 의미하도록 한다. 고로 도둑맞은 배열 중 0인 인덱스나 여분이 있는 배열 중 0인 인덱스가 있다면 continue를 이용해 아래 실행문을 건너뛰도록 한다.

//그리고 마지막으로 +1 이나 -1자리인 숫자를 찾아서 학생 수에 +1 더해주도록 한다.

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
/도둑맞은 학생들과 여분이 있는 학생들의 배열을 또 다른 배열을 만들어 각 각 담는다
    var newreserve = reserve
    var newlost = lost
//학생 수
    var studentnumber = n - lost.count
 
//여분이 있는 학생이자 동시에 도둑맞은 학생
    for i in 0...reserve.count-1{
        for k in 0...lost.count-1{
        if newlost[k] == newreserve[i] {
            newreserve[i] = 0
            newlost[k] = 0
            studentnumber = studentnumber + 1
            break
        }
        }
    }
        
        for j in 0...newlost.count-1{
            for i in 0...newreserve.count-1 {
//받은 학생 또 받는 거 방지
            if newreserve[i] == 0 || newlost[j] == 0 {
              continue
//그리고 마지막으로 +1 이나 -1자리인 숫자를 찾아서 학생 수에 +1 더해주도록 한다.
            }else if newlost[j] - 1 == newreserve[i] || newlost[j] + 1 == newreserve[i]{
                        newreserve[i] = 0
                        newlost[j] = 0
                studentnumber += 1
                break
                    }
                }
            }
    return studentnumber
}

 

 

나의 처음 풀이는 이거였다. 딕셔너리를 활용해서 풀어보려 했지만 자꾸만 테스트케이스 5번 7번에서 틀렸다고 나온다. 모든 테스트케이스를 대입해봐도 틀린 게 없는데 뭐가 틀린지 모르겠다.... 

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
    var student : Set<Int> = []
    var reservedic = [Int:Int]()
    var lostdic = [Int:Int]()
    var studentnumber = 0
    studentnumber += n - lost.count
    
    for i in 0...lost.count-1{
        lostdic[lost[i]] = 0
    }
    print(lostdic)
    
    for i in 0...reserve.count-1{
        reservedic[reserve[i]] = 1
    }
    let sortreserve = reservedic.keys.sorted()
    print(sortreserve)
    for i in 0...reserve.count-1{
        for k in 0...lost.count-1{
        if lost[k] == reserve[i] {
            reservedic[reserve[i]] = 0
            lostdic[lost[k]] = 1
            studentnumber = studentnumber + 1
        }
        }
        
        for j in 0...lost.count-1{
            if reservedic[reserve[i]] == 1{
                if reserve[i] == lost[j]-1{
                    if lostdic[lost[j]] == 0 {
                student.insert(lost[j])
                reservedic[reserve[i]] = 0
                lostdic[lost[j]] = 1
 
                    }
                    }else if lost[j]+1 == reserve[i]{
                        if lostdic[lost[j]] == 0{
                        student.insert(lost[j])
                        reservedic[reserve[i]] = 0
                        lostdic[lost[j]] = 1
                        }
                    }
                }
            }
       
    }
    print(lostdic,reservedic,student)
    
     studentnumber += student.count
    return studentnumber
}