Use python3.10 pattern matching

1 files changed, 7 insertions, 6 deletions

diff --git a/2015/21/puzzles.py b/2015/21/puzzles.py
index 48e8c4e..d355c77 100644
--- a/2015/21/puzzles.py
+++ b/2015/21/puzzles.py
@@ -38,12 +38,13 @@ def wins(d: int, hp: int) -> bool:
 def valid(e: tuple[tuple[str, int, int, int], ...]) -> bool:
 	w = a = r = 0
 	for i in e:
-		if i[0] == "Weapon":
-			w += 1
-		elif i[0] == "Armor":
-			a += 1
-		elif i[0] == "Ring":
-			r += 1
+		match i[0]:
+			case "Weapon":
+				w += 1
+			case "Armor":
+				a += 1
+			case "Ring":
+				r += 1
 	return (w == 1) and (a <= 1) and (r <= 2)