SRM 675 Div2 250

LengthUnitCalculator.py

# -*- coding: utf-8 -*-
import math,string,itertools,fractions,heapq,collections,re,array,bisect

class LengthUnitCalculator:
    def calc(self, amount, fromUnit, toUnit):
        if fromUnit == toUnit:
            return amount

        if fromUnit == "in":
            if toUnit == "ft":
                return amount / 12
            elif toUnit == "yd":
                return amount / (12 * 3)
            elif toUnit == "mi":
                return amount / (12 * 3 * 1760)
        if fromUnit == "ft":
            if toUnit == "in":
                return amount * 12
            elif toUnit == "yd":
                return amount / 3
            elif toUnit == "mi":
                return amount / (3 * 1760)
        if fromUnit == "yd":
            if toUnit == "in":
                return amount * 3 * 12
            elif toUnit == "ft":
                return amount * 3
            elif toUnit == "mi":
                return amount / 1760
        if fromUnit == "mi":
            if toUnit == "in":
                return amount * 12 * 3 * 1760
            elif toUnit == "ft":
                return amount * 3 * 1760
            elif toUnit == "yd":
                return amount * 1760
        return 0.0