PAT B1051 複數乘法 (15分)

題目鏈接：https://pintia.cn/problem-sets/994805260223102976/problems/994805274496319488

題目描述
複數可以寫成 (A+Bi) 的常規形式，其中 A 是實部，B 是虛部，i 是虛數單位，滿足 i​2​​ =−1；也可以寫成極座標下的指數形式 (R×e​(Pi)​​ )，其中 R 是複數模，P 是輻角，i 是虛數單位，其等價於三角形式 R(cos§+isin§)。

現給定兩個複數的 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

代碼

#include <iostream>
#include <cmath>
using namespace std;
int main(){
	double r1, p1, r2, p2;
	cin >> r1 >> p1 >> r2 >> p2;
	double a = (r1 * r2) * cos(p1 + p2);
	double b = (r1 * r2) * sin(p1 + p2);
	if(fabs(a) < 0.01) 
		a = 0;
	if(fabs(b) < 0.01)
		b = 0;
	if(b < 0)
		printf("%.2lf-%.2lfi", a, fabs(b));
	else 
		printf("%.2lf+%.2lfi", a, b);
	return 0;
}