level0_조건에 맞게 수열 변환하기 3

2024. 11. 11. 22:29파이썬_알고리즘(코딩테스트)/level0

 

내가 제출한 코드

def solution(arr, k):
    answer = []
    for i in arr:
        if k % 2 == 1:
            answer.append(i * k)
        else:
            answer.append(i + k)
    return answer
  • if k % 2 == 1:
    • k가 홀수라면 i * k 값을 answer 배열에 추가한다.
  • k가 짝수라면 i + k값을 answer 배열에 추가한다. 

다른 사람이 제출한 코드

def solution(arr,k):
    return [i*k if k%2!=0 else i+k for i in arr]

리스트 컴프리헨션을 이용하여 나타낸 코드이다.

위에 내가 제출한 코드의 설명과 동일하다. 

'파이썬_알고리즘(코딩테스트) > level0' 카테고리의 다른 글

level0_0 떼기  (0) 2024.11.12
level0_특정 문자 제거하기  (0) 2024.11.12
level0_접미사 배열  (1) 2024.10.30
level0_가까운 1 찾기  (0) 2024.10.30
level0_뒤에서 5등까지  (0) 2024.10.29