[백준] 1076 파이썬(python) : 저항

1076번 : 저항

 

 

import sys
color = { 'black' : 0, 'brown' : 1, 'red' : 2, 'orange' : 3, 'yellow' : 4, 'green' : 5, 'blue' : 6, 
          'violet' : 7, 'grey' : 8, 'white' : 9 }
number = { 0 : 1, 1 : 10, 2 : 100, 3 : 1000, 4 : 10000, 5 : 100000, 6 : 1000000, 7 : 10000000, 
           8 : 100000000, 9 : 1000000000 }
arr = [ sys.stdin.readline().rstrip() for _ in range(3) ]
res = 0
res = (color[arr[0]]*10 + color[arr[1]]) * number[color[arr[2]]]
print(res)

 

문제에서 요구하는것이 첫번째원소와 두번째 원소를 더하고 세번째 원소는 곱하기만 하면되기 때문에 해시를 2개 만들어서 처리했다.