PAT 1051 复数乘法 python

2019-04-14 09:09发布

1051 复数乘法 (15 分)

复数可以写成 (A+Bi) 的常规形式,其中 A 是实部,B 是虚部,i 是虚数单位,满足 i​2​​=−1;也可以写成极坐标下的指数形式 (R×e​(Pi)​​),其中 R 是复数模,P 是辐角,i 是虚数单位,其等价于三角形式 (R(cos( P )+isin( P )))。 现给定两个复数的 R 和 P,要求输出两数乘积的常规形式。 输入格式: 输入在一行中依次给出两个复数的 R​1​​, P​1​​, R​2​​, P​2​​,数字间以空格分隔。 输出格式: 在一行中按照 A+Bi 的格式输出两数乘积的常规形式,实部和虚部均保留 2 位小数。注意:如果 B 是负数,则应该写成 A-|B|i 的形式。 输入样例:
2.3 3.5 5.2 0.4
输出样例:
-8.68-8.23i
作者: CHEN, Yue
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB 问题分析:
先粘一段运算法则
在极坐标下,复数可用模长r与幅角θ表示为(r,θ)。对于复数a+bi,r=√(a²+b²),θ=arctan(b/a)。此时,复数相乘表现为幅角相加,模长相乘。
——百度百科
剩下就简单了,后两个测试点会出现问题,是因为python中用round四舍五入,0也是带有符号的,所以需要将0的符号删除 代码: from math import cos, sin r1, p1, r2, p2 = [float(i) for i in input().split()] a = r1 * r2 * cos(p1 + p2) b = r1 * r2 * sin(p1 + p2) a = round(a, 2) b = round(b, 2) if a == 0: a = 0 if b == 0: b = 0 print('{:.2f}{:+.2f}i'.format(a, b))