LeetCode #13. 罗马数字转整数

力扣 | # 13.罗马数字转整数

题目截图

首先利用字典将罗马数字与数字对应起来

romaNumberMap = {
            "I" : 1,
            "V" : 5,
            "X" : 10,
            "L" : 50,
            "C" : 100,
            "D" : 500,
            "M" : 1000,
        }

然后利用enumerate()函数将需要研究的字符串str遍历，使其每个元素有一个下标索引。

enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列，同时列出数据和数据下标，一般用在 for 循环当中。

Python enumerate() 函数 | 菜鸟教程

思路：罗马数字如果左边的值比右边的小，则该值作为负数累加。如果左边的值比右边的大或者相等，则直接累加。

如“X I V”，X=10, I = 1, V = 5。

                                X>I,    I<V,

所以结果等于          +10      -1       +5    =14。

如“MCMXCIV”，M=1000, C=100, M=1000, X=10, C=100, I=1, V=5。

                               M>C,       C<M,        M>X,         X<C,     C>I,         I<V。

            结果等于  +1000         -100        +1000        -10       +100        -1           +5   =  1994

最后一位不用比较直接累加。

class Solution:
    def romanToInt(self, s: str) -> int:
        romaNumberMap = {
            "I" : 1,
            "V" : 5,
            "X" : 10,
            "L" : 50,
            "C" : 100,
            "D" : 500,
            "M" : 1000,
        }
        n = len(s)
        sum =0
        for i, ch in enumerate(s):
            value = romaNumberMap[ch]
            if i < n-1 and value < romaNumberMap[s[i+1]]:
                sum -= value
            elif i < n-1 and value >= romaNumberMap[s[i+1]]:
                sum +=value
        sum +=value
        return sum


if __name__ == "__main__":
    a = Solution()
    print(a.romanToInt("MCMXCIV"))

代码优化

语句判断完后不用elif再判断，直接用else囊括其他情况。

class Solution:
    def romanToInt(self, s: str) -> int:
        romaNumberMap = {
            "I" : 1,
            "V" : 5,
            "X" : 10,
            "L" : 50,
            "C" : 100,
            "D" : 500,
            "M" : 1000,
        }
        n = len(s)
        sum =0
        for i, ch in enumerate(s):
            value = romaNumberMap[ch]
            if i < n-1 and value < romaNumberMap[s[i+1]]:
                sum -= value
            else:
                sum +=value
        return sum


if __name__ == "__main__":
    a = Solution()
    print(a.romanToInt("MCMXCIV"))

也可以不用enumerate()函数

该方法没有用in，所以循环后还是要单独加上sum += romaNumberMap[s[i+1]]

class Solution:
    def romanToInt(self, s: str) -> int:
        romaNumberMap = {
            "I" : 1,
            "V" : 5,
            "X" : 10,
            "L" : 50,
            "C" : 100,
            "D" : 500,
            "M" : 1000,
        }
        n = len(s)
        sum =0
        for i in range(0, n-1):
            if romaNumberMap[s[i]] < romaNumberMap[s[i+1]]:
                sum -= romaNumberMap[s[i]]
            else:
                sum +=romaNumberMap[s[i]]
        sum += romaNumberMap[s[i+1]]
        return sum


if __name__ == "__main__":
    a = Solution()
    print(a.romanToInt("MCMXCIV"))

扩展

面试中可能遇到要自己实现enumerate()函数

def my_enumerate(datas):
    for i in range(len(datas)):
        yield i, datas[i]

Python3 迭代器与生成器 | 菜鸟教程

在 Python 中，使用了 yield 的函数被称为生成器（generator）。.

跟普通函数不同的是，生成器是一个返回迭代器的函数，只能用于迭代操作，更简单点理解生成器就是一个迭代器。

在调用生成器运行的过程中，每次遇到 yield 时函数会暂停并保存当前所有的运行信息，返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行。

调用一个生成器函数，返回的是一个迭代器对象。

 参见Python yield 使用浅析 | 菜鸟教程

yield 的作用：把一个函数变成一个 generator，带有 yield 的函数不再是一个普通函数，Python 解释器会将其视为一个 generator，调用函数f(x)时，不会执行f函数，而是返回一个iterable对象。在 for 循环执行时，每次循环都会执行 fab 函数内部的代码，执行到 yield b 时，f函数就返回一个迭代值，下次迭代时，代码从 yield xxx 的下一条语句继续执行，而函数的本地变量看起来和上次中断执行前是完全一样的，于是函数继续执行，直到再次遇到 yield。

一个带有 yield 的函数就是一个 generator，它和普通函数不同，生成一个 generator 看起来像函数调用，但不会执行任何函数代码，直到对其调用 next()（在 for 循环中会自动调用 next()）才开始执行。虽然执行流程仍按函数的流程执行，但每执行到一个 yield 语句就会中断，并返回一个迭代值，下次执行时从 yield 的下一个语句继续执行。看起来就好像一个函数在正常执行的过程中被 yield 中断了数次，每次中断都会通过 yield 返回当前的迭代值。

yield 的好处是显而易见的，把一个函数改写为一个 generator 就获得了迭代能力，比起用类的实例保存状态来计算下一个 next() 的值，不仅代码简洁，而且执行流程异常清晰。