PAT 乙级 1051 复数乘法 (15分)

1051 复数乘法 (15分)

复数可以写成 (A+Bi) 的常规形式，其中 A 是实部，B 是虚部，i 是虚数单位，满足 i2=−1；也可以写成极坐标下的指数形式 (R×e(Pi))，其中 R 是复数模，P 是辐角，i 是虚数单位，其等价于三角形式 R(cos§+isin§)。
现给定两个复数的 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
作者: CHEN, Yue
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB

这里在使用%.2f进行截断前两位的时候，有形成-0.00的情况，需要矫正，如果不进行处理，测试点3、4过不去

#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
int main(){
    double r_1,p_1,r_2,p_2,a,b;
    cin>>r_1>>p_1>>r_2>>p_2;
    a = r_1 * cos(p_1) * r_2 * cos(p_2) - r_1 * sin(p_1) * r_2 * sin(p_2);
    b = r_1 * cos(p_1) * r_2 * sin(p_2) + r_1 * sin(p_1) * r_2 * cos(p_2);
    if(abs(a)<0.01)
        a=0.00;
    if(abs(b)<0.01)
        b=0.00;
    if(b>=0)
        printf("%.2f+%.2fi",a,b);
    else
        printf("%.2f%.2fi",a,b);
    return 0;
}