From e6c5de02e120a48f6c02a0cd46f9809eafa832ac Mon Sep 17 00:00:00 2001 From: bumpsoo Date: Thu, 15 Aug 2024 14:44:59 +0900 Subject: [PATCH] https://leetcode.com/problems/lemonade-change --- easy/lemonade_change.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 easy/lemonade_change.py diff --git a/easy/lemonade_change.py b/easy/lemonade_change.py new file mode 100644 index 0000000..c9a93d8 --- /dev/null +++ b/easy/lemonade_change.py @@ -0,0 +1,24 @@ +# https://leetcode.com/problems/lemonade-change +from typing import List + +class Solution: + def lemonadeChange(self, bills: List[int]) -> bool: + inventory: List[int] = [0, 0] + for bill in bills: + match bill: + case 5: + inventory[0] += 1 + case 10: + inventory[0] -= 1 + inventory[1] += 1 + if inventory[0] < 0: + return False + case 20: + if inventory[1] == 0: + inventory[0] -= 3 + else: + inventory[1] -= 1 + inventory[0] -= 1 + if inventory[0] < 0 or inventory[1] < 0: + return False + return True