PTA 1034 有理数四则运算(Python3)

本题要求编写程序，计算 2 个有理数的和、差、积、商。

输入格式：

输入在一行中按照 a1/b1 a2/b2 的格式给出两个分数形式的有理数，其中分子和分母全是整型范围内的整数，负号只可能出现在分子前，分母不为 0。

输出格式：

分别在 4 行中按照 有理数1 运算符 有理数2 = 结果 的格式顺序输出 2 个有理数的和、差、积、商。注意输出的每个有理数必须是该有理数的最简形式 k a/b，其中 k 是整数部分，a/b 是最简分数部分；若为负数，则须加括号；若除法分母为 0，则输出 Inf。题目保证正确的输出中没有超过整型范围的整数。

样例-1：">样例-1：">输入样例 1：

2/3 -4/2

输出样例 1：

2/3 + (-2) = (-1 1/3)
2/3 - (-2) = 2 2/3
2/3 * (-2) = (-1 1/3)
2/3 / (-2) = (-1/3)

输入样例 2：

5/3 0/6

输出样例 2：

1 2/3 + 0 = 1 2/3
1 2/3 - 0 = 1 2/3
1 2/3 * 0 = 0
1 2/3 / 0 = Inf

提交结果：

代码：

import fractions

a, b = map(fractions.Fraction, input().split())


def format1(data):
    if '/' not in str(data) or abs(data) < 1:
        if data < 0:
            data = '(' + str(data) + ')'
        return data
    else:
        c, d = str(abs(data)).split('/')
        if data < 0:
            data2 = '(-' + str(int(c) // int(d)) + ' ' + str(int(c) % int(d)) + '/' + d + ')'
        else:
            data2 = str(int(c) // int(d)) + ' ' + str(int(c) % int(d)) + '/' + d
        return data2


print('{} + {} = {}'.format(format1(a), format1(b), format1(a + b)))
print('{} - {} = {}'.format(format1(a), format1(b), format1(a - b)))
print('{} * {} = {}'.format(format1(a), format1(b), format1(a * b)))
if b == 0:
    print('{} / 0 = Inf'.format(format1(a)))
else:
    print('{} / {} = {}'.format(format1(a), format1(b), format1(a / b)))