Python solutions Challenge 080 LK

1 files changed, 34 insertions, 0 deletions

diff --git a/challenge-080/lubos-kolouch/python/ch-2.py b/challenge-080/lubos-kolouch/python/ch-2.py
new file mode 100644
index 0000000000..1496b407d5
--- /dev/null
+++ b/challenge-080/lubos-kolouch/python/ch-2.py
@@ -0,0 +1,34 @@
+#!/bin/env pythoon
+"""
+#===============================================================================
+#
+#         FILE: ch_2.py
+#
+#        USAGE: ./ch_2.py
+#
+#  DESCRIPTION: https://perlweeklychallenge.org/blog/perl-weekly-challenge-080/
+#               Challenge #2
+#               Count Candies
+#
+#       AUTHOR: Lubos Kolouch
+#===============================================================================
+"""
+
+
+def get_candle_count(arr: list):
+    """ Count the candles """
+
+    # We need to give 1 candy to everyone
+    count = len(arr)
+
+    # and then find out number of unique elements as they will be certainly
+    # bigger than neighbor... -1 (the initial poor one)
+
+    count += len(set(arr))
+    count -= 1
+
+    return count
+
+
+assert get_candle_count([1, 2, 2]) == 4
+assert get_candle_count([1, 4, 3, 2]) == 7