3.21 PAT1051 乙级 python

复数乘法

import math
r1,p1,r2,p2 = list(map(float,input().split()))
a = r1*r2*math.cos(p1+p2)
b = r1*r2*math.sin(p1+p2)
if abs(a)<0.01:
	a = 0
if abs(b)<0.01:
	b = 0
if b>=0:
	print('%.2f'%(a)+'+'+'%.2f'%(b)+'i')
else:
	print('%.2f'%(a)+'-'+'%.2f'%(abs(b))+'i')

（抄作业选手→ →）