[코딩테스트] 코딩 기초 트레이닝 / 조건에 맞게 수열 변환하기 3

코딩 기초 트레이닝 👾

https://school.programmers.co.kr/learn/courses/30/lessons/181835

📖 문제

정수 배열 arr와 자연수 k가 주어집니다.

만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더합니다.

이러한 변환을 마친 후의 arr를 return 하는 solution 함수를 완성해 주세요.

💻 답안

def solution(arr, k):
    answer = []
    if(k%2==1):
        for i in arr:
            answer.append(i*k)

    else:
        for i in arr:
            answer.append(i+k)
    return answer