CF1118D2 Coffee and Coursework (Hard Version)(二分)

本文最后更新于:3 years ago

题目描述


Input

The first line of the input contains two integers n and m (1≤n≤2⋅105, 1≤m≤109) — the number of cups of coffee and the number of pages in the coursework.

The second line of the input contains n integers a1,a2,…,an (1≤ai≤109), where ai is the caffeine dosage of coffee in the i-th cup.

Output
If it is impossible to write the coursework, print -1. Otherwise print the minimum number of days Polycarp needs to do it.

Examples
input

1
2
5 8
2 3 1 1 2

output

1
4

input

1
2
7 10
1 3 4 2 1 4 2

output

1
2

input

1
2
5 15
5 5 5 5 5

output

1
1

input

1
2
5 16
5 5 5 5 5

output

1
2

input

1
2
5 26
5 5 5 5 5

output

1
-1

题目大意:
有n杯咖啡,要做m份作业,每杯咖啡有一个值a,表示喝完能做的作业数量,但是如果一天之内喝了超过一杯,那么每喝一杯就会少做k-1件作业(即第一杯为ai,第二杯为max(0,ai-1),第n杯为max(0,ai-n)),问最少多少天能做完作业

解题思路:
要使咖啡收益最大化,那么就得使值大的在每天作为第一杯咖啡使用,那么可以二分答案,每次取一个天数mid,跑for遍历咖啡,每mid杯咖啡后更新减的值(每天一杯,那么每mid天之后的就是要减去值的),最后输出最少的天数即可,如果咖啡的值加起来也无法超过作业的数量的话,那么显然是无法完成的,输出-1即可

题外话:
python的效率是比较慢的,在打acm过程中是很吃亏的,如果oj有pypy选项的,建议使用pypy提交,pypy虽然库支持比较少,但是运行效率会比python快很多

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class Solution():
def __init__(self):
firstline=input().split()
n=int(firstline[0])
m=int(firstline[1])
a=input().split()
for i in range(n):
a[i]=int(a[i])
a.sort(key=int,reverse=True)#快排,让值高的咖啡在前面
l=1
r=m+1
while l<r:
mid=int(l+r>>1)
sum=0
for i in range(n):
if a[i]-int(i/mid)<=0:#剪枝,无法获得值了就没必要再运算了
break
sum+=max(0,a[i]-int(i/mid))#第一个mid范围表示每天的第一杯,第二个mid范围表示第二杯,即要减去的值是i/mid
if sum>=m: #同理,做完了作业就不需要再喝咖啡了
break
if sum>=m:
r=mid
else:
l=mid+1
if r==m+1:
print('-1')
else:
print(r)



if __name__ == '__main__':
Solution()


本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!