Solutions CS101

Appearance

03406: 书架 ​

greedy, http://cs101.openjudge.cn/practice/03406/

John最近买了一个书架用来存放奶牛养殖书籍，但书架很快被存满了，只剩最顶层有空余。

John共有N头奶牛(1 ≤ N ≤ 20,000)，每头奶牛有自己的高度Hi(1 ≤ Hi ≤ 10,000)，N头奶牛的总高度为S。书架高度为B(1 ≤ B ≤ S < 2,000,000,007).

为了到达书架顶层，奶牛可以踩着其他奶牛的背，像叠罗汉一样，直到他们的总高度不低于书架高度。当然若奶牛越多则危险性越大。为了帮助John到达书架顶层，找出使用奶牛数目最少的解决方案吧。

输入

第1行：空格隔开的整数N和B 第2~N+1行：第i+1行为整数Hi

输出

能达到书架高度所使用奶牛的最少数目

样例输入

6 40
6
18
11
13
19
11

样例输出

3
python
n, B = map(int, input().split())
h = []
for _ in range(n):
    h.append(int(input()))

h.sort(reverse=True)
i = 0
s = 0
while i <= n:
    s += h[i]
    i += 1
    if s >= B:
        break
    
print(i)

二分查找

python
# 蒋子轩23工学院
def min_cows_to_reach(N, B):
	# 二分查找变形，找大于等于B的最小索引
    left, right = 1, N
    while left < right:  #注意不能取等
        mid = (left + right) // 2 #左偏
        if prefix_sum[mid]>=B:  #等于时继续向左找
            right = mid   #注意不-1，
        else:
            left = mid + 1
    return left  #return不取等的那个
N, B = map(int, input().split())
cows = [int(input()) for _ in range(N)]
#优先选择高的
cows.sort(reverse=True)
#计算前缀和
prefix_sum = [0] * (len(cows) + 1)
for i in range(1, len(cows)+1):
    prefix_sum[i] = prefix_sum[i-1] + cows[i-1]
print(min_cows_to_reach(N, B))