#!/usr/bin/env python
# coding=utf-8

##########################################################
# Leetcode 13 罗马数字转整数
#
# 罗马数字包含以下七种字符: I， V， X， L，C，D 和 M。
#   字符          数值
#   I             1
#   V             5
#   X             10
#   L             50
#   C             100
#   D             500
#   M             1000
#
# 例如， 罗马数字 2 写做 II ，即为两个并列的 1。12 写做 XII ，即为 X + II 。 27 写做  XXVII, 即为 XX + V + II 。
# 
# 通常情况下，罗马数字中小的数字在大的数字的右边。但也存在特例，例如 4 不写做 IIII，而是 IV。数字 1 在数字 5 的左边，所表示的数等于大数 5 减小数 1 得到的数值 4 。
# 同样地，数字 9 表示为 IX。这个特殊的规则只适用于以下六种情况：
#   - I 可以放在 V (5) 和 X (10) 的左边，来表示 4 和 9。
#   - X 可以放在 L (50) 和 C (100) 的左边，来表示 40 和 90。 
#   - C 可以放在 D (500) 和 M (1000) 的左边，来表示 400 和 900。
#
# 给定一个罗马数字，将其转换成整数。输入确保在 1 到 3999 的范围内。
#
# 示例 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.
##########################################################


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

class Solution:
    def romanToInt(self, s: str) -> int:
        """
        
        (Knowledge)

        思路：
        1. 首先用map记录罗马字符和对应数值的映射关系；
        2. 接着每遍历一个字符，就将其对应的值加到结果里面；
        3. 每轮遍历的时候，如果是 'I', 'X', 'C' 字符，则判断其右边是否有能和其搭配使用的特殊值，如果有则减去两倍的对应数值
        """
        result = 0
        length = len(s)
        for i in range(length):
            result += myMap[s[i]]
            if s[i] == 'I' and i < length - 1 and (s[i + 1] == 'V' or s[i + 1] == 'X'):
                result -= (2 * 1)
            if s[i] == 'X' and i < length - 1 and (s[i + 1] == 'L' or s[i + 1] == 'C'):
                result -= (2 * 10)
            if s[i] == 'C' and i < length - 1 and (s[i + 1] == 'D' or s[i + 1] == 'M'):
                result -= (2 * 100)
        return result


if __name__ == '__main__':
    solution = Solution()
    print(solution.romanToInt("III"), "= 3")
    print(solution.romanToInt("IV"), "= 4")
    print(solution.romanToInt("IX"), "= 9")
    print(solution.romanToInt("LVIII"), "= 58")
    print(solution.romanToInt("MCMXCIV"), "= 1994")