LeetCode-Complex Number Multiplication

最新推荐文章于 2021-08-25 20:26:42 发布

BeHelium

最新推荐文章于 2021-08-25 20:26:42 发布

阅读量156

点赞数

分类专栏： LeetCode 文章标签： LeetCode Java

本文链接：https://blog.csdn.net/qq_24133491/article/details/82559064

版权

LeetCode 专栏收录该内容

306 篇文章 0 订阅

订阅专栏

Description：
Given two strings representing two complex numbers.
You need to return a string representing their multiplication. Note i2 = -1 according to the definition.

Example 1:

Input: "1+1i", "1+1i"
Output: "0+2i"

Explanation: (1 + i) * (1 + i) = 1 + i2 + 2 * i = 2i, and you need convert it to the form of 0+2i.

Example 2:

Input: "1+-1i", "1+-1i"
Output: "0+-2i"

Explanation: (1 - i) * (1 - i) = 1 + i2 - 2 * i = -2i, and you need convert it to the form of 0+-2i.

Note:

The input strings will not have extra blank.
The input strings will be given in the form of a+bi, where the integer a and b will both belong to the range of [-100, 100]. And the output should be also in this form.

题意：给定两个用字符串表示的复数，计算复数乘积；

解法：要计算出两个复数的乘积，我们需要分离出两个复数的实部和虚部，并且保存各自的虚部符号；最后计算时我们由两个数的实部乘积+虚部乘积（i²=-1）得到结果的实部；由一个数的实部与另一个数的虚部乘积得到结果的虚部；最后将结果转换为字符串表示；要注意的是如果实部为0也需要表示出来；

class Solution {
    public String complexNumberMultiply(String a, String b) {
        StringBuilder result = new StringBuilder();
        int firstSymbol = 1;
        int secondSymbol = 1;
        int firstReal, firstImg, secondReal, secondImg;
        int firstImgIndex, secondImgIndex;
        int resultReal, resultImg;

        firstReal = Integer.valueOf(a.substring(0, a.indexOf('+')));
        secondReal = Integer.valueOf(b.substring(0, b.indexOf('+')));
        firstImgIndex = a.indexOf('+') + 1;
        secondImgIndex = b.indexOf('+') + 1;
        if (a.charAt(a.indexOf('+') + 1) == '-') {firstSymbol *= -1; firstImgIndex++;}
        if (b.charAt(b.indexOf('+') + 1) == '-') {secondSymbol *= -1; secondImgIndex++;}
        firstImg = Integer.valueOf(a.substring(firstImgIndex, a.length() - 1));
        secondImg = Integer.valueOf(b.substring(secondImgIndex, b.length() - 1));

        resultReal = firstReal * secondReal + firstImg * secondImg * firstSymbol * secondSymbol * -1;
        resultImg = firstReal * secondImg * secondSymbol + secondReal * firstImg * firstSymbol;

        result.append(resultReal);
        result.append("+");
        result.append(resultImg);
        result.append("i");

        return result.toString();
    }

}