1
1
mirror of https://github.com/theoludwig/programming-challenges.git synced 2024-12-08 00:45:29 +01:00
programming-challenges/challenges/roman-numerals/solutions/python/function/solution.py

60 lines
1.7 KiB
Python

from typing import TypedDict
import sys
class RomanArabicMatch(TypedDict):
arabic: int
roman: str
matches_roman_arabic: list[RomanArabicMatch] = [
{'arabic': 1000, 'roman': 'M'},
{'arabic': 900, 'roman': 'CM'},
{'arabic': 500, 'roman': 'D'},
{'arabic': 400, 'roman': 'CD'},
{'arabic': 100, 'roman': 'C'},
{'arabic': 90, 'roman': 'XC'},
{'arabic': 50, 'roman': 'L'},
{'arabic': 40, 'roman': 'XL'},
{'arabic': 10, 'roman': 'X'},
{'arabic': 9, 'roman': 'IX'},
{'arabic': 5, 'roman': 'V'},
{'arabic': 4, 'roman': 'IV'},
{'arabic': 1, 'roman': 'I'}
]
def convert_arabic_to_roman(arabic_number: int) -> str:
roman_number: str = ''
for match in matches_roman_arabic:
while arabic_number >= match['arabic']:
roman_number = roman_number + match['roman']
arabic_number = arabic_number - match['arabic']
return roman_number
def convert_roman_to_arabic(roman_number: str) -> int:
arabic_number: int = 0
for match in matches_roman_arabic:
while True:
try:
if roman_number.index(match['roman']) == 0:
arabic_number = arabic_number + match['arabic']
roman_number = roman_number.replace(match['roman'], '', 1)
else:
break
except ValueError:
break
return arabic_number
input_values: list[str] = []
for value in sys.stdin:
input_values.append(value.rstrip('\n'))
convert_type = input_values[0]
if convert_type == 'arabic to roman':
print(convert_arabic_to_roman(int(input_values[1])))
else:
print(convert_roman_to_arabic(input_values[1]))