AcWing 4818. 奶牛大学

AcWing 4818. 奶牛大学

原题

Farmer John 计划为奶牛们新开办一所大学！

有 N 头奶牛可能会入学。

每头奶牛最多愿意支付 ci 的学费。

Farmer John 可以设定所有奶牛入学需要支付的学费。

如果这笔学费大于一头奶牛愿意支付的最高金额，那么这头奶牛就不会入学。

Farmer John 想赚尽可能多的钱，从而可以给他的讲师提供一笔可观的工资。

请求出他能赚到的钱的数量，以及此时应当收取多少学费。

输入格式

输入的第一行包含 N。

第二行包含 N个整数 c1,c2,…,cN1,2,…,，其中 ci 是奶牛 i 愿意支付的最高学费金额。

输出格式

输出 Farmer John 可以赚到的最大金额以及最优情况下他应该收取的学费。如果有多个解，输出收取学费最小的解。

注意这个问题涉及到的整数可能需要使用 6464 位整数型（例如，Java 中的 “long”，C/C++ 中的 “long long”）。

数据范围

1≤N≤105,
1≤ci≤106。

输入样例：

4
1 6 4 6

输出样例：

12 4

样例解释

如果 Farmer John 收费 44，那么 33 头奶牛将会入学，从而使他赚取 3×4=123×4=12 的金额。

题解

# AcWing 4818. 奶牛大学
N=int(input())
costs_s=input().split(' ')
costs_n=[int(cos) for cos in costs_s]
# 从大到小排列
costs_n.sort(reverse=True)

cossums=[]
xuefeis=[]
cossum=0
qian=0
for i in range(N-1,-1,-1):
    if qian==costs_n[i] or costs_n[i]==0:
        continue
    tmp=costs_n[i]*(i+1)
    if cossum>=tmp:
        cossums.append(cossum)
        xuefeis.append(qian)
    else:
        cossum=tmp
    qian=costs_n[i]

maxcos=max(cossums)
maxcosid=cossums.index(maxcos)
print(cossums[maxcosid],xuefeis[maxcosid])

• 1.我去搜了一下python封装的sort函数的时间复杂度是nlogn，所以sort＋遍历一次＋查找cossums*2和xuefeis*1不会超时
• 2.最高收益且最低学费对应的学费应该是理想学费之一：如果低于理想学费，说明可以继续往上提高也会有同样的一批奶牛来报名，那可以提高来增大收益
• 3.排序后当当前奶牛的价格和前面遍历过的奶牛价格一致时可以直接跳过了，因为奶牛数少了但是单价不变，（这里的单价指收的学费，奶牛价格指奶牛的最高学费）
• 4.最开始想的是，如果前面就已经出现了一个局部极值，就直接跳出循环，但是毕竟是局部的，1234，可能3比24都高，但是可能1一个就可以顶其他几个了，所以这里也会出现局部极值，所以就设置了一个cossums来存放局部最优解
• 5.大于等于的放入局部最优解中的原因：
  • 大于则说明前面那个是局部最优
  • 等于则说明这两可能都是局部最优，把较小的一个放进去（满足题目中’‘如果收益一样取最低学费’‘）