537. 复数乘法

最新推荐文章于 2025-03-14 12:19:15 发布

原创最新推荐文章于 2025-03-14 12:19:15 发布 · 160 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #java

每日刷题专栏收录该内容

302 篇文章

订阅专栏

该博客主要讨论了如何实现复数乘法的算法，给定两个复数字符串表示，如'1+1i'和'1+1i'，通过解析字符串并进行数学运算，得到乘积的字符串表示，例如'0+2i'。代码中展示了详细的解析和计算过程。

537. 复数乘法https://leetcode.cn/problems/complex-number-multiplication/

难度中等140

复数可以用字符串表示，遵循 "实部+虚部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 的形式。

提示：

num1 和 num2 都是有效的复数表示。

通过次数41,403提交次数55,319

class Solution {
    public String complexNumberMultiply(String num1, String num2) {

        int flag = 0;
        int af1=1,a1=0,af2=1,a2=0;
        for(int i=0;i<num1.length();i++)
        {
            if(num1.charAt(0)=='-') af1 = -1;
            if(num1.charAt(i)>='0' && num1.charAt(i)<='9' && flag==0)
            {
                a1 = a1*10+(num1.charAt(i)-'0');
            }
            else if(i>0 && num1.charAt(i)=='-' )
            {
                af2 = -1;
                flag =1;
            } 
            else if(i>0 && num1.charAt(i)=='+' ) flag = 1;
            else if(num1.charAt(i)>='0' && num1.charAt(i)<='9' && flag==1)
            {
                a2 = a2*10+(num1.charAt(i)-'0');
            }
        }
        a1 = a1*af1;
        a2 = a2*af2;

        flag = 0;
        int bf1=1,b1=0,bf2=1,b2=0;
        for(int i=0;i<num2.length();i++)
        {
            if(num2.charAt(0)=='-') bf1 = -1;
            if(num2.charAt(i)>='0' && num2.charAt(i)<='9' && flag==0)
            {
                b1 = b1*10+(num2.charAt(i)-'0');
            }
            else if(i>0 && num2.charAt(i)=='-' )
            {
                bf2 = -1;
                flag =1;
            } 
            else if(i>0 && num2.charAt(i)=='+' ) flag = 1;
            else if(num2.charAt(i)>='0' && num2.charAt(i)<='9' && flag==1)
            {
                b2 = b2*10+(num2.charAt(i)-'0');
            }
        }
        b1 = b1*bf1;
        b2 = b2*bf2;


        int a = a1*b1;
        int ab = a1*b2+b1*a2;
        int b = a2*b2*(-1);
        return (a+b)+"+"+(ab+"i");
    }
}