본문 바로가기
Coding Test/프로그래머스 코딩테스트 입문

Python 짝수는 싫어요 [코딩테스트 입문 - Day3]

by giem 2023. 1. 16.
반응형

 

 


문제


구현

list comprehension과 range를 이용해 한 step씩 건너뛰도록 구현하면 된다.


코드
def solution(n):
    return [i for i in range(1, n+1, 2)]

range (시작, 끝, 스텝) 이렇게 사용된다.

위 코드는 1이상 n+1 미만 범위에서 2씩 건너뛰도록 한 것이다.

 

 

즉 [1,1+2,1+2+2,..... ] 이렇게 n까지 증가하게 된다.


다른 풀이
def solution(n):
    return [x for x in range(n + 1) if x % 2]

range의 스텝값을 빼고 % 2의 나머지가 있을 때(==홀수일 때)를 추출해 줘도 된다.

반응형

댓글