[99클럽 코테 스터디 30일차 TIL] 상자넣기 - 백준 1965

동적계획법인데 이전 문제 너무 똑같다!

 

https://www.acmicpc.net/problem/1965

정육면체 모양의 상자가 일렬로 늘어서 있다. 
상자마다 크기가 주어져 있는데, 앞에 있는 상자의 크기가 뒤에 있는 상자의 크기보다 작으면, 
앞에 있는 상자를 뒤에 있는 상자 안에 넣을 수가 있다. 

예를 들어 앞에서부터 순서대로 크기가 (1, 5, 2, 3, 7)인 5개의 상자가 있다면, 
크기 1인 상자를 크기 5인 상자에 넣고, 다시 이 상자를 크기 7인 상자 안에 넣을 수 있다. 
하지만 이렇게 상자를 넣을 수 있는 방법은 여러 가지가 있을 수 있다. 
앞의 예에서 차례대로 크기가 1, 2, 3, 7인 상자를 선택하면 총 4개의 상자가 한 개의 상자에 들어가게 된다.

상자의 크기가 주어질 때, 한 번에 넣을 수 있는 최대의 상자 개수를 출력하는 프로그램을 작성하시오.

[입력]
파일의 첫 번째 줄은 상자의 개수 n (1 ≤ n ≤ 1000)을 나타낸다. 
두 번째 줄에는 각 상자의 크기가 순서대로 주어진다. 
상자의 크기는 1,000을 넘지 않는 자연수이다.

[출력]
첫째 줄에 한 줄에 넣을 수 있는 최대의 상자 개수를 출력한다.

풀이 접근

LIS랑 똑같은 문제이다.

단 요소를 구하는 것이 아니라 길이만 구하는 것이기 때문에 이진탐색을 활용했다.

var lis: [Int] = []
for num in nums {
    if lis.isEmpty || lis.last! < num {
        lis.append(num)
    } else {
        var left = 0, right = lis.count - 1
        while left < right {
            let mid = (left + right) / 2
            if lis[mid] < num {
                left = mid + 1
            } else {
                right = mid
            }
        }
        lis[left] = num
    }
}

최종 코드 - Swift

_ = readLine()
let nums = readLine()!.split(separator: " ").map{ Int("\($0)")! }
var lis: [Int] = []
for num in nums {
    if lis.isEmpty || lis.last! < num {
        lis.append(num)
    } else {
        var left = 0, right = lis.count - 1
        while left < right {
            let mid = (left + right) / 2
            if lis[mid] < num {
                left = mid + 1
            } else {
                right = mid
            }
        }
        lis[left] = num
    }
}
print(lis.count)

Conclusion

이전이랑 너무 비슷한(거의 똑같은) 문제라서 조금 아쉽다.