結果

問題 No.518 ローマ数字の和
コンテスト
ユーザー Schnee
提出日時 2019-09-25 17:13:33
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 32 ms / 2,000 ms
コード長 1,106 bytes
コンパイル時間 92 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-09-22 08:59:09
合計ジャッジ時間 1,581 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

n = int(input().strip())
rom_nums = input().strip().split()
rom_dic = {"I":1, "V": 5, "X":10, "L":50, "C":100, "D":500, "M":1000}

ara_nums = []
for i in rom_nums:
    for j in range(len(i) - 1):
        if rom_dic[i[j]] >= rom_dic[i[j + 1]]:
            ara_nums.append(rom_dic[i[j]])
        else:
            ara_nums.append(-(rom_dic[i[j]]))
    ara_nums.append(rom_dic[i[-1]])
ara_sum = sum(ara_nums)
ara_sum_str = str(ara_sum)

if ara_sum > 3999:
    print("ERROR")
    sys.exit(0)

def get_key(a_num):
    return [k for k, v in rom_dic.items() if v == a_num][0]

rom_sum = []
for i in range(1,len(ara_sum_str) + 1):
    digit = 10 ** (i - 1)
    if int(ara_sum_str[-i]) == 4:
        rom_sum.append(get_key(digit) + get_key(5 * digit))
    elif int(ara_sum_str[-i]) == 9:
        rom_sum.append(get_key(digit) + get_key(10 * digit))
    elif 0 <= int(ara_sum_str[-i]) <= 3:
        rom_sum.append(get_key(digit) * int(ara_sum_str[-i]))
    else:
        rom_sum.append(get_key(5 * digit) + get_key(digit) * (int(ara_sum_str[-i]) - 5))
rom_sum = reversed(rom_sum)

print("".join(rom_sum))
0