礼物费用最少流利说编程题-------python

公司准备给n名用户快递礼物,在甲地和乙地各有个仓库存放了a和b个这种礼物。从甲地和乙地给每名用户快递礼物的费用是不同的,请计算让N名用户都收到礼物的最低费用是多少
样例输入:
3
1 2
13 19
4 9
10 20
样例输出:
38

import sys
import heapq
number=int(sys.stdin.readline().strip().split()[0])
str=sys.stdin.readline().strip().split()
a=int(str[0])
b=int(str[1])
i=0
arr=[]
while ia:
    result = []
    for i in range(0, number):
        brr.append(arr[i][1] - arr[i][0])
        result = map(brr.index, heapq.nsmallest(b, brr))
    #print result
    for i in range(0,number):
        if i in result:
            sum+=arr[i][1]
        else:
            sum+=arr[i][0]
else:
    result = []
    for i in range(0, number):
        brr.append(arr[i][0] - arr[i][1])
        result = map(brr.index, heapq.nsmallest(a, brr))
    #print result
    for i in range(0,number):
        if i in result:
            sum+=arr[i][0]
        else:
            sum+=arr[i][1]
print sum

你可能感兴趣的:(面试)