PAT~乙级~1051 复数乘法 ~python-蒲公英云

PAT~乙级~1051 复数乘法 ~python

题目描述：

复数可以写成 (A+Bi) 的常规形式，其中 A 是实部，B 是虚部，i 是虚数单位，满足 i2=−1；也可以写成极坐标下的指数形式 (R×e(Pi))，其中 R 是复数模，P 是辐角，i 是虚数单位，其等价于三角形式 (R(cos(P)+isin(P))。

现给定两个复数的 R 和 P，要求输出两数乘积的常规形式。

输入格式：

输入在一行中依次给出两个复数的 R1, P1, R2, P2，数字间以空格分隔。

输出格式：

在一行中按照 A+Bi 的格式输出两数乘积的常规形式，实部和虚部均保留 2 位小数。注意：如果 B 是负数，则应该写成 A-|B|i 的形式。

输入样例：

2.3 3.5 5.2 0.4

输出样例：

-8.68-8.23i

-————————————————————————————-答题区—————————————————————————————

注意python的一个坑，不要使用round来确定保留几位小数，如果结果为1.0，round保留两位小数也为1.0，别在这坑掉就行。

import math
def sanjiao_changgui(R,P):
    A=R*math.cos(P)
    B=R*math.sin(P)
    return A,B
s=list(map(float,input().split()))
A1,B1=sanjiao_changgui(s[0],s[1])
A2,B2=sanjiao_changgui(s[2],s[3])
A3=A1*A2-B1*B2
B3=A1*B2+A2*B1
if(round(A3,2)!=0):
    print("%.2f"%A3,end='')
else:
    print("0.00",end='')
if(round(B3,2)==0):
    print("+0.00i")
elif(B3<0):
    print("%.2fi"%B3)
else:
    print("+%.2fi"%B3)