[Leetcode] Complex Number Multiplication (#537)

 

 

문제 

 

A complex number can be represented as a string on the form "real+imaginaryi" where:

• real is the real part and is an integer in the range [-100, 100].
• imaginary is the imaginary part and is an integer in the range [-100, 100].
• i2 == -1.

Given two complex numbers num1 and num2 as strings, return a string of the complex number that represents their multiplications.

 

 

Example 1:

Input: num1 = "1+1i", num2 = "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: num1 = "1+-1i", num2 = "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.

 

 

해설 코드 

class Solution(object):
    def complexNumberMultiply(self, a, b):
        
        
        # Split numbers to real and imagine parts without "i".
        ra, ia = a[:-1].split("+")
        rb, ib = b[:-1].split("+")

        # Convert all parts to integer.
        ra, rb = int(ra), int(rb)
        ia, ib = int(ia),int(ib)

        # Return result with formula of multiply for complex numbers.
        return "{}+{}i".format((ra*rb)-(ia*ib),(ra*ib)+(ia*rb))

분석 > 이 문제에 접근하지 못했던 것은 어떻게 문자열을 수식으로 만들어야 할지 감을 잡지 못했기 때문이다. 이 코드는 우선 복소수의 정수 부분과 복소수가 아닌 정수 부분을 분리하여 계수만 따로 저장하여 수식을 계산한 후에 문자열 형식으로 만들어줌으로써 문제를 해결하였다. 계수만을 따로 분리할 때는 복소수 i 는 무조건 문자열의 가장 끝에 위치하므로 [:-1]을 이용하여 마지막 문자 i를 제외하고 문자열을 슬라이싱 해주었고 +를 기준으로 문자열을 split한 것은 +가 없어도 기본적으로 각 계수는 정수형태를 띄기 때문이다. 

 

 

 

Complex Number Multiplication - LeetCode