python实现罗马数字转换为整数

题目描述

上一篇文章中，我们介绍了将阿拉伯数字转换为罗马数字的方法，里面详细介绍了罗马数字的书写规则，请参考python实现将整数转换为罗马数字

LeetCode原题地址：https://leetcode-cn.com/problems/roman-to-integer/

测试用例

• 示例 1

输入: “III”
输出: 3

• 示例 2

输入: “IV”
输出: 4

• 示例 3

输入: “IX”
输出: 9

• 示例 4

输入: “LVIII”
输出: 58
解释: L = 50, V= 5, III = 3.

• 示例 5

输入: “MCMXCIV”
输出: 1994
解释: M = 1000, CM = 900, XC = 90, IV = 4.

代码实现

其实实现的思路很简单，我们就先建立罗马数字与阿拉伯数字对应的字典，然后再通过遍历罗马数字字符串，遍历时比较左边和右边对应数字的大小，如果右边大于左边再转换的时候就需要利用两位进行转换否则就只使用一位即可，实现代码如下

class Solution:
    def romanToInt(self, s: str) -> int:
        #定义罗马数字转阿拉伯数字的字典
        roman_to_int = {"M": 1000, "CM": 900, "D": 500, "CD": 400, "C": 100,
                        "XC": 90, "L": 50, "XL": 40, "X": 10, "IX": 9, "V": 5,
                        "IV": 4, "I": 1}
        res_int = 0
        s_len = len(s)
        i = 0
        while i < s_len-1:
            if roman_to_int[s[i]] < roman_to_int[s[i+1]]:
                res_int += roman_to_int[s[i:i+2]]
                i += 2
            else:
                res_int += roman_to_int[s[i]]
                i += 1
        if i == s_len - 1:
            res_int += roman_to_int[s[i]]

        return res_int

solution = Solution()
print(solution.romanToInt("MCMXCIV"))

除此之外，还有一种比较简洁的写法，每次转换的时候都先转换一个2位字符的罗马数字，如果找不到对应的就转换一位，因为在进行2位字符转换的时候会导致前面一位被重复计算，所以在定义字典的时候，我们先减去左边罗马数字对应的数字值，避免重复计算的问题。

class Solution:
    def romanToInt(self, s: str) -> int:
        #定义罗马数字转阿拉伯数字的字典
        #需要注意的是这里的字典定义的数字要比实际的小一个左边的数字
        roman_to_int = {"M": 1000, "CM": 800, "D": 500, "CD": 300, "C": 100,
                        "XC": 80, "L": 50, "XL": 30, "X": 10, "IX": 8, "V": 5,
                        "IV": 3, "I": 1}
        #这里的max(i-1,0)防止下标越界
        return sum(roman_to_int.get(s[max(i-1,0):i+1],roman_to_int[s_c]) for i,s_c in enumerate(s))

参考：

1. Leetcode详细题解