[python] Code-kata week3-1

🖥 Code-kata week3-1 문제

문제
두 개의 input에는 복소수(complex number)가 string 으로 주어집니다. 복소수란 a+bi 의 형태로, 실수와 허수로 이루어진 수입니다.

input으로 받은 두 수를 곱해서 반환해주세요. 반환하는 표현도 복소수 형태의 string 이어야 합니다.

복소수 정의에 의하면 (i^2)는 -1 이므로 (i^2) 일때는 -1로 계산해주세요.

제곱 표현이 안 되어 i의 2제곱을 (i^2)라고 표현했습니다.

가정: input은 항상 a+bi 형태입니다. output도 a+bi 형태로 나와야 합니다.

Input: "1+1i", "1+1i" Output: "0+2i" 설명: (1 + i) * (1 + i) = 1 + i + i + i^2 = 2i 2i를 복소수 형태로 바꾸면 0+2i.

Input: "1+-1i", "1+-1i" Output: "0+-2i" 설명: (1 - i) * (1 - i) = 1 - i - i + i^2 = -2i, -2i를 복소수 형태로 바꾸면 0+-2i.

풀이

인자, 반환하는 표현도 string 형태여야 한다.
그래서 slicing을 바로 적용해서 계산하는 방식을 사용했다.

• 반환하는 형태도 무조건 a+bi 형태다.
• a는 '+' 전까지 모든 문자열에 해당한다. (int로 변환)
• b는 '+' 다음부터 'i' 전까지의 문자열에 해당한다. (int로 변환)
• i^2는 -1로 변환한다

def complex_number_multiply(a, b):
  x = int(a[:a.index('+')])
  p = int(b[:b.index('+')])
  y = int(a[a.index('+')+1:a.index('i')])
  q = int(b[b.index('+')+1:b.index('i')])
  
  result1 = (x*p - y*q)
  result2 = (x*q + y*p)
  if result1 == 0: result1 = 0
  if result2 == 0: result2 = 0
  
  # '+'로 연결하려면 모두 str 형태여야 함 
  return str(result1) + '+' + str(result2) + 'i'