每日一题 --- 537. 复数乘法[力扣][Go]

func complexNumberMultiply(num1 string, num2 string) string {
    // 先获取实部数值
    var a1,a2,b1,b2 int
    f := func(c rune) bool {
        if c == '+' || c == 'i'{
            return true
        } else {
            return false
        }
    }
    one := strings.FieldsFunc(num1, f)
    two := strings.FieldsFunc(num2, f)
    a1,_ = strconv.Atoi(one[0])
    a2,_ = strconv.Atoi(one[1])
    b1,_ = strconv.Atoi(two[0])
    b2,_ = strconv.Atoi(two[1])
    // 计算
    aa := a1 * b1 - a2 * b2
    bb := a1 * b2 + a2 * b1
    return strconv.Itoa(aa) + "+" + strconv.Itoa(bb) + "i"
}