c_cpp 537.复数乘法

Posted 2021-05-14

struct ComplexInt {
    int real;
    int imag;
    
    ComplexInt(string c) {
        int ppos = 0;
        while(c[ppos] != '+') ppos += 1;
        real = stoi(c.substr(0, ppos));
        imag = stoi(c.substr(ppos + 1, c.size() - ppos - 2)); // start after '+' and end before 'i'
    }
    
    ComplexInt(int r, int i) : real(r), imag(i) {}
    
    string ToString() {
        return to_string(real) + "+" + to_string(imag) + "i";
    }
    
    ComplexInt operator *(const ComplexInt& rhs) const {
        return ComplexInt((real * rhs.real) - (imag * rhs.imag), (real * rhs.imag) + (imag * rhs.real));
    }
};

class Solution {
public:
    string complexNumberMultiply(string a, string b) {
        return (ComplexInt(a) * ComplexInt(b)).ToString();
    }
};