算法题15 安排机器 牛客网,腾讯笔试题
算法题15 安排机器 牛客网,腾讯笔试题
链接:https://www.nowcoder.com/questionTerminal/42e7ff5c5696445ab907caff17fc9e15
来源:牛客网
小Q的公司最近接到m个任务, 第i个任务需要xi的时间去完成, 难度等级为yi。
小Q拥有n台机器, 每台机器最长工作时间zi, 机器等级wi。
对于一个任务,它只能交由一台机器来完成, 如果安排给它的机器的最长工作时间小于任务需要的时间, 则不能完成,如果完成这个任务将获得200 * xi + 3 * yi收益。
对于一台机器,它一天只能完成一个任务, 如果它的机器等级小于安排给它的任务难度等级, 则不能完成。
小Q想在今天尽可能的去完成任务, 即完成的任务数量最大。如果有多种安排方案,小Q还想找到收益最大的那个方案。小Q需要你来帮助他计算一下。
输入描述:
输入包括N + M + 1行,
输入的第一行为两个正整数n和m(1 <= n, m <= 100000), 表示机器的数量和任务的数量。
接下来n行,每行两个整数zi和wi(0 < zi < 1000, 0 <= wi <= 100), 表示每台机器的最大工作时间和机器等级。
接下来的m行,每行两个整数xi和yi(0 < xi < 1000, 0 <= yi<= 100), 表示每个任务需要的完成时间和任务的难度等级。
输出描述:
输出两个整数, 分别表示最大能完成的任务数量和获取的收益。
示例1
输入
1 2 100 3 100 2 100 1
输出
1 20006
解题代码:
# -*- coding:utf-8 -*- #python2.7 n,m=[int(i) for i in raw_input().strip().split()] machines,tasks=[],[] for i in range(n): hours,level=[int(i) for i in raw_input().strip().split()] machines.append([hours,level]) for i in range(m): hours,level=[int(i) for i in raw_input().strip().split()] tasks.append([hours,level]) machines.sort(key=lambda x: (x[0],x[1]),reverse=True) tasks.sort(key=lambda x: (x[0],x[1]),reverse=True) dp=[0 for i in range(101)] j=0 count=0 res=0 for tktime,tklev in tasks: while j<len(machines) and machines[j][0]>=tktime: dp[machines[j][1]]+=1 j+=1 for i in range(tklev,101): if dp[i]>0: dp[i]-=1 res += 200*tktime+3*tklev count+=1 break print("%d %d" %(count,res))