diff --git a/src/codewars/kata_chinese_numeral_encoder.py b/src/codewars/kata_chinese_numeral_encoder.py index 816043b..de4746a 100644 --- a/src/codewars/kata_chinese_numeral_encoder.py +++ b/src/codewars/kata_chinese_numeral_encoder.py @@ -30,11 +30,20 @@ D = {2: 10, 3: 100, 4: 1000, 5: 10000} def to_chinese_numeral(n): + # Check if n in C dictonary try: return C[n] except KeyError: pass + # Check if negative + if n < 0: + n = abs(n) + is_negative = True + else: + is_negative = False + + # Put single digits in list split_number = [int(x) for x in str(n)] arranged_number = [] @@ -46,7 +55,10 @@ def to_chinese_numeral(n): if arranged_number[-1] == 0: arranged_number.pop(-1) + if is_negative: + arranged_number.append("-") + return arranged_number -print(to_chinese_numeral(120)) +print(to_chinese_numeral(-90909)) # 9 10000 9