Python 코드카타: 수박 | 내적

수박수박수박수박수박수?

길이가 n이고, "수박수박수박수...."와 같은 패턴을 유지하는 문자열을 리턴하는 함수, solution을 완성하세요. 예를들어 n이 4이면 "수박수박"을 리턴하고 3이라면 "수박수"를 리턴하면 됩니다.

 

# 내 풀이
def solution(n):
    answer = ''
    if n % 2 == 0:            #짝수일 경우
        answer = '수박'*(n//2)
    else:                     #홀수일 경우
        answer = '수박'*(n//2) + '수'
    return answer
    
# 다른 풀이
def solution(n):
    answer = ''
    for i in range(1, n+1):
        if i % 2 == 0:      #짝수일 경우
            answer += "박"
        else:               #홀수일 경우
            answer += "수"
    return answer

내적

길이가 같은 두 1차원 정수 배열 a, b가 매개변수로 주어집니다. a와 b의 내적을 return 하도록 solution 함수를 완성해주세요.

이때, a와 b의 내적은 a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1] 입니다. (n은 a, b의 길이)

def solution(a, b):
    answer = 0
    for i, j in zip(a, b):
        answer += i * j
    return answer

• 두 리스트에서 원소를 불러오기 위해 zip 함수 사용