python 練習題- 圖書租金計算

題目：

每個用戶初始金額300

1、扣除規則
1）圖書價格大於等於 100元，累計借出去天數小於等於15天，每本每天租金5元，累計超過15天的，超過的天數每天每本租金3元
2）圖書價格大於等於 50元、小於100元，累計借出去天數小於等於15天，每本每天租金3元，累計超過15天的，超過的天數每天每本租金2元
3）圖書價格小於 50元，每本每天租金1元

2、余額小於所借圖書價格，不能借；可以繼續借閱其他的書

3、租金最大不超過所借圖書的價格

4、超期還書，每超過一天，另外扣1元

輸出余額

例子：
[[130,5,11],[60,16,16],[10,11,11]]

輸出：
182

解釋：
130,5,11   實際11天，超期6天，則租金為 11*5+6 = 61
60,16,16   實際11天，有一天超過15天，則租金為 15*3 + 2 = 47
10,11,11   實際11天，11*1 = 11，大於圖書價格10元，則租金為10
總共為 61+47+10 = 118，余額 182

 

代碼：

# @Author  :whyCai
# @Time    :2021/2/22 9:44


from typing import List
"""
計算剩余金額
"""
class Solution:
    def chargeBalance(self, histList: List[List[int]]) -> int:
        initBalance = 300

        histListLen = len(histList)
        for i in range(histListLen):
            #圖書價格小於余額
            if histList[i][0] < initBalance:
                #判斷圖書價格再哪個區間
                if histList[i][0] >=100:
                    #實際借書日期是否再15天內
                    chargeBalance = histList[i][2] * 5 if histList[i][2] <= 15 else 15 * 5 + (histList[i][2] -15) * 3
                elif histList[i][0] >=50 and histList[i][0] <100:
                    chargeBalance = histList[i][2] * 3 if histList[i][2] <= 15 else 15 * 3 + (histList[i][2] -15) * 2
                else:
                    chargeBalance = histList[i][2] * 1
                # 判斷是否超期還書
                if histList[i][2] > histList[i][1]:
                    chargeBalance = chargeBalance + (histList[i][2] - histList[i][1]) * 1
                #判斷最大租金不超過所借圖書價格
                if chargeBalance > histList[i][0]:
                    chargeBalance = histList[i][0]
                initBalance = initBalance - chargeBalance

        return initBalance

print(Solution().chargeBalance([[130,5,11],[60,16,16],[10,11,11]]))