2120 纸张尺寸 Python

描述：

在ISO国际标准中定义了A0纸张的大小为1189mm×841mm,将A0纸沿长边对折后为A1纸，大小为841mm×594mm，在对折的过程中长度直接取下整(实际裁剪时可能有损耗)。将A1纸沿长边对折后为 A2 纸,依此类推。输入纸张的名称，请输出纸张的大小。

输入：

输入一行包含一个字符串表示纸张的名称，该名称一定是A0、A1、A2、A3、A4、A5、A6、A7、A8、A9之一。

输出：

输出两行，每行包含一个整数，依次表示长边和短边的长度。

样例输入：

A0

样例输出： 

1189 841

 基于python实现的代码：

str_input = input()
n = int(str_input[1:])
longth = 1189
wideth = 841
if n == 0:
    longth = 1189
    wideth = 841
else:
    for i in range(n):
        longth, wideth = wideth, longth//2
print(longth)
print(wideth)