箱の中のボールの得点を変更できる場合における期待値の最小化

前提・実現したいこと

箱の中にN(Nは自然数)種類の色のボールが入っています。また, N色のボールが何個ずつ入っているかはリストAで与えられています。(リストAはすべて自然数)
（例）A =[1,1,1]なら1色目が1個, 2色目が3個, 3色目が3個含まれています。　　そして, i(1≦i≦N)色目のボールを取り出すと獲得できる得点がリストBで与えられています。(リストBはすべて自然数)
（例）B=[1,2,3]　1色目のボールを引くと1点, 2色目のボールを引くと2点, 3色目のボールを引くと3点を獲得できます。
このとき, K個だけ箱から取り出す前にボールの色を任意の色に変えることができます。
（例）K = 1のとき, 1個のボールの色を変えることができ, 3色目のボールを1色目の色に変えることができます。すると, A = [2,1,0]となります。
これらの条件において, ボールを1個取り出すときに取りうる最小の期待値を求めるという問題です。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

python3
1
2def ball(N,K,A,B):
3    C = sorted(B,reverse=True)
4    counter = 0
5    if K > len(A):
6        K = len(A)
7
8    while True:
9        a = B.index(C[counter]) #n番目に得点の高いボールを探す
10        b = B.index(min(B)) 
11        if K <= A[a]:  #ある1色のボールの色を全て変えられない場合
12            A[b] = A[b] + K
13            A[a] = A[a] - K
14            K = 0 #K個だけ色を変え,色を変えられるボールの個数を0個とする
15
16        elif K > A[a]: #ある1色のボールの色を全て変えられる場合
17            A[b] = A[b] + A[a]
18            A[a] = 0
19            K = K - A[a] #Kから色を変えたボールの個数だけ値を減じる
20        if K <= 0: #色を変えられなくなった場合, ループを終了
21            break
22        else: #まだ色が変えられる場合, 次に得点の高い色を探す
23            counter += 1
24
25    ans = 0
26    for i in range(N):
27        ans += (A[i]*B[i]) 
28
29    return ans/sum(A)
30
31#例を代入しています
32N = 3
33K = 1
34A = [1, 1, 1]
35B = [1, 2, 3]
36print( ball(N, K, A, B) )
37

試したこと

方針としては　期待値が最小⇔リストAとBの内積が最小　となると考えています。K個のボールの色を変えられるので, Bの値が大きく(=得られる得点が高く)なる色のボールを優先して減らし, 最小得点しか得られない色に変えていきます。
色を変えたらKの値から色を変えたボールの数を引き, 2番目に得られる得点が大きい色のボールを減らしていきます。このようにして高い得点の得られる色のボールを減らしていき, 1個もボールの色が変えられなくなったら繰り返しを終了させています。最後に, 最小化したリストA, Bの内積を取り, ボールの総個数で割ることで期待値を求めています。