def calc(a, b):
    res = a // b + a % b * 2 + (a - b) ** 2
    return res