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 |