[Python]조건에 맞게 수열 변환하기3

<문제 설명>

정수 배열 arr와 자연수 k가 주어집니다.만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더합니다.이러한 변환을 마친 후의 arr를 return 하는 solution 함수를 완성해 주세요.

입출력 예 설명

입출력 예 #1
주어진 k인 3은 홀수이므로, 전체 배열에 3을 곱합니다. 따라서 [3, 6, 9, 300, 297, 294]을 return 합니다.

입출력 예 #2
주어진 k인 2는 짝수이므로, 전체 배열에 2를 더합니다. 따라서 [3, 4, 5, 102, 101, 100]을 return 합니다.

 

def solution(arr, k):
    #홀수를 판별한다. 
    if k % 2 == 1:
       #arr 인덱스 매기기
        for i in range(len(arr)):
            #arr의 원소에 k를 곱한다. 
            arr[i] = arr[i] *k
    else:
        for i in range(len(arr)):
            arr[i] = arr[i] +k
        
    return arr

 

<keypoint>

• 나는 배열 안의 원소들에 특정 산술 혹은 논리 연산 후 다시 해당 배열을 반환하는 문제에 대해 그동안 arr.append()함수를 사용하곤 했다. 하지만 해당 방법은 기존의 배열을 그대로 출력하고 새로운 값을 붙여 반환해주는 요소 추가하는 리스트의 내장함수이다. 

#append 사용할 경우
num_list = [1,23,4,5,6,7,8,9,99,44]

num_list.append(24)

num_list
#예상 출력 : [1,23,4,5,6,7,8,9,99,44,24]

• 하지만 이젠 배열의 원소에 특정 연산을 취한 후 다시 배열에 넣는 과정이 아주 기초적이고 단순한 "할당"과정을 통해 할 수 있다는 점을 깨달았다. 
• 이때 인덱스를 이용하는 것이 좋다. 
• 예를 들어, arr[i] = arr[i] * k