PTA 1034 有理数四则运算 (20 分)

题目

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

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

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

输入样例 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
结尾无空行

解题思路

inputArr = input().split()
# inputArr = "2/3 -4/2".split()

from fractions import Fraction
A1, A2 = map(int, inputArr[0].split("/"))
B1, B2 = map(int, inputArr[1].split("/"))
f1 = Fraction(A1, A2)
f2 = Fraction(B1, B2)
def format1(num):

    if '/' not in str(num) or abs(num)<1:
        if num<0:
            num = '('+ str(num)+')'
        return num
    else:
        n1, n2 = map(int, str(abs(num)).split('/'))
        if num < 0:
            res = '(-' + str(n1//n2) + ' ' + str(n1%n2)+'/'+str(n2)+')'
        else:
            res = str(n1 // n2) + ' ' + str(n1 % n2) + '/' + str(n2)
        return res
print('{} + {} = {}'.format(format1(f1), format1(f2), format1(f1+f2)))
print('{} - {} = {}'.format(format1(f1), format1(f2), format1(f1-f2)))
print('{} * {} = {}'.format(format1(f1), format1(f2), format1(f1*f2)))
if f2 == 0:
    print("{} / {} = {}".format(format1(f1), format1(f2), "Inf"))
else:
    print("{} / {} = {}".format(format1(f1), format1(f2), format1(f1/f2)))