python实现贪婪算法的找零钱问题

import numpy as np
import math
def main():
    money=[0.01,0.05,0.1,0.5,1,5]
    num=input("请输入每种面额纸币的张数： ")
    num=list(map(int,num.split(' ')))
    need=int(input("请输入所要找零钱的金额："))
    res=[]
    #计算收营员一共的钱
    s=sum(np.array(money)*np.array(num))
    i=5
    if s<need:
        print("余额不足")
    else:
        while need>0:
            d=math.floor(need/money[i])
            if d>num[i]:
                d=num[i]
            res.append(d)
            need-=d*money[i]
            print("需要%d张%d元纸币"%(d,money[i]))
            i = i - 1

if  __name__ == '__main__':
    main()