プレゼント代は最低流暢にプログラミング問題を話します------python

会社はn人のユーザーに宅配便の贈り物を用意し、甲地と乙地にそれぞれaとb個の贈り物を倉庫に保管した.甲地と乙地から各ユーザーにプレゼントを宅配する費用は異なります.N名のユーザーにプレゼントをもらう最低費用は何サンプルですか.入力:3 1 2 13 19 4 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