LeetCode 537. 复数乘法

537. 复数乘法
     复数 可以用字符串表示，遵循 “实部+虚部i” 的形式，并满足下述条件：

实部 是一个整数，取值范围是 [-100, 100]
虚部 也是一个整数，取值范围是 [-100, 100]
i2 == -1
给你两个字符串表示的复数 num1 和 num2 ，请你遵循复数表示形式，返回表示它们乘积的字符串。

示例 1：
输入：num1 = “1+1i”, num2 = “1+1i”
输出：“0+2i”
解释：(1 + i) * (1 + i) = 1 + i2 + 2 * i = 2i ，你需要将它转换为 0+2i 的形式。
示例 2：
输入：num1 = “1±1i”, num2 = “1±1i”
输出：“0±2i”
解释：(1 - i) * (1 - i) = 1 + i2 - 2 * i = -2i ，你需要将它转换为 0±2i 的形式。

注意题中输入格式必定是1±1i这种，肯定会出现加号。

class Solution {
   public  String complexNumberMultiply(String num1, String num2) {
            int n1 =num1.length();
            int n2 =num2.length();
            int head =0;
            int f=0;//有+的时候为1
            int flag =1;
            //num1的两个数
            int a=0;
            int b=0;
            //num2的两个数
            int x=0;
            int y=0;
            for(int i=0;i<n1;i++){
                if(num1.charAt(i) == '+'){
                    a=Integer.parseInt(num1.substring(head,i));
                    f=1;
                    if(num1.charAt(i+1) != '-'){
                        head=i+1;
                    }

                }
                if(num1.charAt(i) == '-'){
                    if(f==1){
                        
                    flag=-1;
                    head=i+1;
                    }
                  
                }
                if(num1.charAt(i) == 'i'){
                    b=Integer.parseInt(num1.substring(head,i))*flag;
                }
            }
            f=0;
            flag=1;
            head=0;
            for(int i=0;i<n2;i++){
                if(num2.charAt(i) == '+'){
                    x=Integer.parseInt(num2.substring(head,i));
                    f=1;
                    if(num2.charAt(i+1) != '-'){
                        head=i+1;
                    }
                }
                if(num2.charAt(i) == '-'){
                    if(f==1){                
                    flag=-1;
                    head=i+1;
                    }
                   
                    
                }
                if(num2.charAt(i) == 'i'){
                    y=Integer.parseInt(num2.substring(head,i))*flag;
                }
            }
          
            int k =a * x+b*y*(-1);//实数部分
            int w =b*x+a*y;//虚数部分
           
             return k+"+"+w+"i";
            
            
        }
}