aboutsummaryrefslogtreecommitdiff
path: root/challenge-214/lubos-kolouch/python
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-214/lubos-kolouch/python')
-rw-r--r--challenge-214/lubos-kolouch/python/ch-1.py31
-rw-r--r--challenge-214/lubos-kolouch/python/ch-2.py43
2 files changed, 74 insertions, 0 deletions
diff --git a/challenge-214/lubos-kolouch/python/ch-1.py b/challenge-214/lubos-kolouch/python/ch-1.py
new file mode 100644
index 0000000000..710c81ea42
--- /dev/null
+++ b/challenge-214/lubos-kolouch/python/ch-1.py
@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+from typing import List
+
+
+def rank_scores(scores: List[int]) -> List[str]:
+ sorted_scores = sorted(scores, reverse=True)
+ ranks = {}
+ rank = 1
+ for i in range(len(sorted_scores)):
+ if i > 0 and sorted_scores[i] == sorted_scores[i - 1]:
+ ranks[sorted_scores[i]] = ranks[sorted_scores[i - 1]]
+ else:
+ ranks[sorted_scores[i]] = rank
+ rank += 1
+ output = []
+ for score in scores:
+ if ranks[score] == 1:
+ output.append('G')
+ elif ranks[score] == 2:
+ output.append('S')
+ elif ranks[score] == 3:
+ output.append('B')
+ else:
+ output.append(str(ranks[score]))
+ return output
+
+
+scores = [1, 2, 4, 3, 5]
+result = rank_scores(scores)
+print(result)
diff --git a/challenge-214/lubos-kolouch/python/ch-2.py b/challenge-214/lubos-kolouch/python/ch-2.py
new file mode 100644
index 0000000000..72d7207b50
--- /dev/null
+++ b/challenge-214/lubos-kolouch/python/ch-2.py
@@ -0,0 +1,43 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+
+def max_score(numbers):
+
+ def helper(numbers, memo):
+ if not numbers:
+ return 0
+
+ key = tuple(numbers)
+ if key in memo:
+ return memo[key]
+
+ max_score = 0
+ i = 0
+ while i < len(numbers):
+ j = i
+ while j < len(numbers) and numbers[j] == numbers[i]:
+ j += 1
+
+ length = j - i
+ next_numbers = numbers[:i] + numbers[j:]
+ score = length * length + helper(next_numbers, memo)
+ max_score = max(max_score, score)
+ i = j
+
+ memo[key] = max_score
+ return max_score
+
+ return helper(numbers, {})
+
+
+# Test cases
+numbers1 = [2, 4, 3, 3, 3, 4, 5, 4, 2]
+numbers2 = [1, 2, 2, 2, 2, 1]
+numbers3 = [1]
+numbers4 = [2, 2, 2, 1, 1, 2, 2, 2]
+
+print(max_score(numbers1)) # Output: 23
+print(max_score(numbers2)) # Output: 20
+print(max_score(numbers3)) # Output: 1
+print(max_score(numbers4)) # Output: 40