From 73b1c7ec14481d7163a6f882f6450fa526813d2d Mon Sep 17 00:00:00 2001
From: Lubos Kolouch <lubos@kolouch.net>
Date: Fri, 24 Nov 2023 09:29:25 +0100
Subject: feat(challenge-244/lubos-kolouch/perl,python,raku,blog): Challenge
 244 LK Perl Python Raku blog

---
 challenge-244/lubos-kolouch/python/ch-2.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 challenge-244/lubos-kolouch/python/ch-2.py

(limited to 'challenge-244/lubos-kolouch/python/ch-2.py')

diff --git a/challenge-244/lubos-kolouch/python/ch-2.py b/challenge-244/lubos-kolouch/python/ch-2.py
new file mode 100644
index 0000000000..2968b5c1d8
--- /dev/null
+++ b/challenge-244/lubos-kolouch/python/ch-2.py
@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+
+import unittest
+from itertools import combinations
+from typing import List
+
+
+def group_hero_power(nums: list[int]) -> int:
+    total_power = 0
+    for i in range(1, len(nums) + 1):
+        for combo in combinations(nums, i):
+            total_power += max(combo) ** 2 * min(combo)
+    return total_power
+
+
+# Tests
+class TestGroupHeroPower(unittest.TestCase):
+    def test_example(self):
+        self.assertEqual(group_hero_power([2, 1, 4]), 141)
+
+
+if __name__ == "__main__":
+    unittest.main()
-- 
cgit